Searched refs:broadcast_deltas (Results 1 – 2 of 2) sorted by relevance
50 bool broadcast_deltas = deltas_in.shape().dims() == 0; in Compute() local57 if (!broadcast_deltas) in_sizes.push_back(deltas_in.shape().dim_size(0)); in Compute()79 T delta = broadcast_deltas ? deltas(0) : deltas(row); in Compute()95 T delta = broadcast_deltas ? deltas(0) : deltas(row); in Compute()
69 const bool broadcast_deltas = NumElements(&input_deltas) == 1; in EvalT() local76 if (!broadcast_deltas) in_sizes.push_back(input_deltas.dims->data[0]); in EvalT()104 const T delta = broadcast_deltas ? deltas[0] : deltas[row]; in EvalT()125 const T delta = broadcast_deltas ? deltas[0] : deltas[row]; in EvalT()