Searched refs:y_batch_indices (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | matmul_op_impl.h | 96 const auto& y_batch_indices = bcast.y_batch_indices(); in Run() local
100 const int64 y_batch_index = should_bcast ? y_batch_indices[i] : i; in Run()
140 const auto& y_batch_indices = bcast.y_batch_indices();
144 const int64 y_batch_index = should_bcast ? y_batch_indices[i] : i;
183 const auto& y_batch_indices = bcast.y_batch_indices();
186 const int64 y_batch_index = should_bcast ? y_batch_indices[i] : i;
386 const std::vector<int64>& b_batch_indices = bcast.y_batch_indices();
559 const std::vector<int64>& b_batch_indices = bcast.y_batch_indices();
|
| D | random_binomial_op.cc | 194 const auto& probs_batch_indices = bcast.y_batch_indices(); in operator ()()
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | matrix_triangular_solve_op_impl.h | 89 const auto& y_batch_indices = bcast.y_batch_indices(); in Run() local
92 const int64 y_batch_index = should_bcast ? y_batch_indices[i] : i; in Run()
281 const std::vector<int64>& b_batch_indices = bcast.y_batch_indices();
|
| D | banded_triangular_solve_op.cc | 67 const auto& y_batch_indices = bcast.y_batch_indices(); in Run() local
73 const int64 y_batch_index = should_bcast ? y_batch_indices[i] : i; in Run()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | matmul_bcast.h | 85 const std::vector<int64>& y_batch_indices() const { return y_batch_indices_; } in y_batch_indices() function
|
| D | matmul_bcast_test.cc | 33 strings::StrAppend(&ret, "[", absl::StrJoin(b.y_batch_indices(), ","), "]"); in MatMulBCastToStr()
|
| D | bcast_test.cc | 51 strings::StrAppend(&ret, "[", absl::StrJoin(b.y_batch_indices(), ","), "]"); in BCastBatchIndices()
|
| D | bcast.h | 396 const std::vector<int64>& y_batch_indices() const { in y_batch_indices() function
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | unroll_batch_matmul.cc | 166 rhs_batch_idx = bcast.y_batch_indices()[batch_idx]; in createMatMulOps()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | unroll_batch_matmul.cc | 219 ? bcast.y_batch_indices()[batch_idx] in Run()
|