Searched defs:expand_spec (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dquantized_instance_norm.cc311 expand_spec; variable
312 expand_spec.set(0, N);
313 expand_spec.set(3, C);
317 const Eigen::array<Index, 4> expand_spec{N, 1, 1, C};
356 float_mean.reshape(expand_spec).broadcast(broadcast_spec))))
363 float_mean.reshape(expand_spec).broadcast(broadcast_spec)) *
366 .reshape(expand_spec)

Completed in 116 milliseconds