Home
last modified time | relevance | path

Searched refs:should_bcast (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmatmul_op_impl.h94 const bool should_bcast = bcast.IsBroadcastingRequired(); in Run() local
99 const int64 x_batch_index = should_bcast ? x_batch_indices[i] : i; in Run()
100 const int64 y_batch_index = should_bcast ? y_batch_indices[i] : i; in Run()
126 const bool should_bcast = bcast.IsBroadcastingRequired();
130 if (batch_size == 1 && !should_bcast) {
143 const int64 x_batch_index = should_bcast ? x_batch_indices[i] : i;
144 const int64 y_batch_index = should_bcast ? y_batch_indices[i] : i;
181 const bool should_bcast = bcast.IsBroadcastingRequired();
185 const int64 x_batch_index = should_bcast ? x_batch_indices[i] : i;
186 const int64 y_batch_index = should_bcast ? y_batch_indices[i] : i;
Drandom_binomial_op.cc192 const bool should_bcast = bcast.IsBroadcastingRequired(); in operator ()() local
207 if (should_bcast) { in operator ()()
Dparameterized_truncated_normal_op.cc359 const bool should_bcast = bcast.IsBroadcastingRequired(); in operator ()() local
380 if (should_bcast) { in operator ()()
/external/tensorflow/tensorflow/core/kernels/linalg/
Dbanded_triangular_solve_op.cc65 const bool should_bcast = bcast.IsBroadcastingRequired(); in Run() local
72 const int64 x_batch_index = should_bcast ? x_batch_indices[i] : i; in Run()
73 const int64 y_batch_index = should_bcast ? y_batch_indices[i] : i; in Run()
Dmatrix_triangular_solve_op_impl.h87 const bool should_bcast = bcast.IsBroadcastingRequired(); in Run() local
91 const int64 x_batch_index = should_bcast ? x_batch_indices[i] : i; in Run()
92 const int64 y_batch_index = should_bcast ? y_batch_indices[i] : i; in Run()