Searched refs:broadcast_spec (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dquantized_instance_norm.cc307 broadcast_spec; variable
308 broadcast_spec.set(1, H);
309 broadcast_spec.set(2, W);
316 const Eigen::array<Index, 4> broadcast_spec{1, H, W, 1};
356 float_mean.reshape(expand_spec).broadcast(broadcast_spec))))
363 float_mean.reshape(expand_spec).broadcast(broadcast_spec)) *
367 .broadcast(broadcast_spec);

Completed in 80 milliseconds