Searched refs:y_batch_size (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | matmul_bcast_test.cc | 44 EXPECT_EQ(4, bcast.y_batch_size()); in TEST()
57 EXPECT_EQ(1, bcast.y_batch_size()); in TEST()
70 EXPECT_EQ(48, bcast.y_batch_size()); in TEST()
83 EXPECT_EQ(6, bcast1.y_batch_size()); in TEST()
92 EXPECT_EQ(1, bcast2.y_batch_size()); in TEST()
122 EXPECT_EQ(4, bcast.y_batch_size()); in TEST()
134 EXPECT_EQ(4, bcast.y_batch_size()); in TEST()
|
| D | matmul_bcast.h | 71 const int64 y_batch_size() const { return y_batch_size_; } in y_batch_size() function
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | matmul_op_impl.h | 348 b_device_memory.reserve(bcast.y_batch_size());
361 std::min(bcast.x_batch_size(), bcast.y_batch_size()) == 1;
367 b_stride = bcast.y_batch_size() != 1 ? k * n : 0;
390 for (int64 i = 0; i < bcast.y_batch_size(); ++i) {
520 b_device_memory.reserve(bcast.y_batch_size());
534 std::min(bcast.x_batch_size(), bcast.y_batch_size()) == 1;
540 b_stride = bcast.y_batch_size() != 1 ? k * n : 0;
563 for (int64 i = 0; i < bcast.y_batch_size(); ++i) {
697 in1_reshaped.CopyFrom(in1, TensorShape({bcast.y_batch_size(), d2, d3})),
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | matrix_triangular_solve_op_impl.h | 188 in1_reshaped.CopyFrom(in1, TensorShape({bcast.y_batch_size(), d2, d3})),
273 bcast.y_batch_size() * m * n * sizeof(Scalar))
282 for (int64 i = 0; i < bcast.y_batch_size(); ++i) {
|
| D | banded_triangular_solve_op.cc | 243 in1_reshaped.CopyFrom(in1, TensorShape({bcast.y_batch_size(), d2, d3})), in Compute()
|
| D | einsum_op_impl.h | 563 TF_RETURN_IF_ERROR(ReshapeToRank3(inputs[1], bcast.y_batch_size(), &rhs)); in ContractOperands()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | unroll_batch_matmul.cc | 288 sliceInput(input_rhs, bcast.y_batch_size(), loc, rewriter); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | unroll_batch_matmul.cc | 206 SliceInput(input_rhs, base_name, "b", bcast.y_batch_size(), input_array_b, in Run()
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend.py | 2078 y_batch_size = y_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:
|