Searched refs:expand_spec (Results 1 – 1 of 1) sorted by relevance
311 expand_spec; in Compute() local312 expand_spec.set(0, N); in Compute()313 expand_spec.set(3, C); in Compute()317 const Eigen::array<Index, 4> expand_spec{N, 1, 1, C}; in Compute() local356 float_mean.reshape(expand_spec).broadcast(broadcast_spec)))) in Compute()363 float_mean.reshape(expand_spec).broadcast(broadcast_spec)) * in Compute()366 .reshape(expand_spec) in Compute()