Searched refs:x_batch_indices (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | matmul_op_impl.h | 95 const auto& x_batch_indices = bcast.x_batch_indices(); in Run() local
99 const int64 x_batch_index = should_bcast ? x_batch_indices[i] : i; in Run()
139 const auto& x_batch_indices = bcast.x_batch_indices();
143 const int64 x_batch_index = should_bcast ? x_batch_indices[i] : i;
182 const auto& x_batch_indices = bcast.x_batch_indices();
185 const int64 x_batch_index = should_bcast ? x_batch_indices[i] : i;
385 const std::vector<int64>& a_batch_indices = bcast.x_batch_indices();
558 const std::vector<int64>& a_batch_indices = bcast.x_batch_indices();
|
| D | random_binomial_op.cc | 193 const auto& counts_batch_indices = bcast.x_batch_indices(); in operator ()()
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | matrix_triangular_solve_op_impl.h | 88 const auto& x_batch_indices = bcast.x_batch_indices(); in Run() local
91 const int64 x_batch_index = should_bcast ? x_batch_indices[i] : i; in Run()
363 const std::vector<int64>& a_batch_indices = bcast.x_batch_indices();
|
| D | banded_triangular_solve_op.cc | 66 const auto& x_batch_indices = bcast.x_batch_indices(); in Run() local
72 const int64 x_batch_index = should_bcast ? x_batch_indices[i] : i; in Run()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | matmul_bcast.h | 80 const std::vector<int64>& x_batch_indices() const { return x_batch_indices_; } in x_batch_indices() function
|
| D | matmul_bcast_test.cc | 32 strings::StrAppend(&ret, "[", absl::StrJoin(b.x_batch_indices(), ","), "]"); in MatMulBCastToStr()
|
| D | bcast.h | 389 const std::vector<int64>& x_batch_indices() const { in x_batch_indices() function
|
| D | bcast_test.cc | 50 strings::StrAppend(&ret, "[", absl::StrJoin(b.x_batch_indices(), ","), "]"); in BCastBatchIndices()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | unroll_batch_matmul.cc | 165 lhs_batch_idx = bcast.x_batch_indices()[batch_idx]; in createMatMulOps()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | unroll_batch_matmul.cc | 216 ? bcast.x_batch_indices()[batch_idx] in Run()
|