Searched refs:y_batch_size (Results 1 – 8 of 8) 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() 118 EXPECT_EQ(4, bcast.y_batch_size()); in TEST() 130 EXPECT_EQ(4, bcast.y_batch_size()); in TEST()
|
D | matmul_bcast.h | 41 const int64 y_batch_size() const { return y_batch_size_; } in y_batch_size() function
|
/external/tensorflow/tensorflow/core/kernels/ |
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 | batch_matmul_op_impl.h | 325 b_device_memory.reserve(bcast.y_batch_size()); 349 for (int64 i = 0; i < bcast.y_batch_size(); ++i) { 461 b_device_memory.reserve(bcast.y_batch_size()); 485 for (int64 i = 0; i < bcast.y_batch_size(); ++i) { 624 in1_reshaped.CopyFrom(in1, TensorShape({bcast.y_batch_size(), d2, d3})),
|
D | einsum_op_impl.h | 580 TF_RETURN_IF_ERROR(ReshapeToRank3(inputs[1], bcast.y_batch_size(), &rhs)); in ContractOperands()
|
/external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
D | unroll_batch_matmul.cc | 205 SliceInput(input_rhs, base_name, "b", bcast.y_batch_size(), input_array_b, in Run()
|
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
D | unroll_batch_matmul.cc | 293 sliceInput(input_rhs, bcast.y_batch_size(), loc, rewriter); in matchAndRewrite()
|
/external/tensorflow/tensorflow/python/keras/ |
D | backend.py | 1764 y_batch_size = y_shape[0] 1766 if x_batch_size is not None and y_batch_size is not None: 1767 if x_batch_size != y_batch_size:
|