Searched refs:broadcast_spec (Results 1 – 1 of 1) sorted by relevance
307 broadcast_spec; in Compute() local308 broadcast_spec.set(1, H); in Compute()309 broadcast_spec.set(2, W); in Compute()316 const Eigen::array<Index, 4> broadcast_spec{1, H, W, 1}; in Compute() local356 float_mean.reshape(expand_spec).broadcast(broadcast_spec)))) in Compute()363 float_mean.reshape(expand_spec).broadcast(broadcast_spec)) * in Compute()367 .broadcast(broadcast_spec); in Compute()