Searched refs:x_batch_size (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | matmul_bcast_test.cc | 43 EXPECT_EQ(1, bcast.x_batch_size()); in TEST()
56 EXPECT_EQ(1, bcast.x_batch_size()); in TEST()
69 EXPECT_EQ(48, bcast.x_batch_size()); in TEST()
82 EXPECT_EQ(1, bcast1.x_batch_size()); in TEST()
91 EXPECT_EQ(2, bcast2.x_batch_size()); in TEST()
121 EXPECT_EQ(3, bcast.x_batch_size()); in TEST()
133 EXPECT_EQ(3, bcast.x_batch_size()); in TEST()
|
| D | matmul_bcast.h | 70 const int64 x_batch_size() const { return x_batch_size_; } in x_batch_size() function
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | matmul_op_impl.h | 347 a_device_memory.reserve(bcast.x_batch_size());
361 std::min(bcast.x_batch_size(), bcast.y_batch_size()) == 1;
366 a_stride = bcast.x_batch_size() != 1 ? m * k : 0;
387 for (int64 i = 0; i < bcast.x_batch_size(); ++i) {
519 a_device_memory.reserve(bcast.x_batch_size());
534 std::min(bcast.x_batch_size(), bcast.y_batch_size()) == 1;
539 a_stride = bcast.x_batch_size() != 1 ? m * k : 0;
560 for (int64 i = 0; i < bcast.x_batch_size(); ++i) {
689 in0_reshaped.CopyFrom(in0, TensorShape({bcast.x_batch_size(), d0, d1})),
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | matrix_triangular_solve_op_impl.h | 180 in0_reshaped.CopyFrom(in0, TensorShape({bcast.x_batch_size(), d0, d1})),
353 a_tmp_ptrs.reserve(bcast.x_batch_size());
364 for (int64 i = 0; i < bcast.x_batch_size(); ++i) {
|
| D | banded_triangular_solve_op.cc | 235 in0_reshaped.CopyFrom(in0, TensorShape({bcast.x_batch_size(), d0, d1})), in Compute()
|
| D | einsum_op_impl.h | 561 TF_RETURN_IF_ERROR(ReshapeToRank3(inputs[0], bcast.x_batch_size(), &lhs)); in ContractOperands()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | unroll_batch_matmul.cc | 286 sliceInput(input_lhs, bcast.x_batch_size(), loc, rewriter); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | unroll_batch_matmul.cc | 203 SliceInput(input_lhs, base_name, "a", bcast.x_batch_size(), input_array_a, in Run()
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend.py | 2077 x_batch_size = x_shape[0]
2080 if x_batch_size is not None and y_batch_size is not None:
2081 if x_batch_size != y_batch_size:
|