Home
last modified time | relevance | path

Searched refs:y_batch_size (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dmatmul_bcast_test.cc44 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()
Dmatmul_bcast.h41 const int64 y_batch_size() const { return y_batch_size_; } in y_batch_size() function
/external/tensorflow/tensorflow/core/kernels/
Dmatrix_triangular_solve_op_impl.h188 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) {
Dbatch_matmul_op_impl.h325 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})),
Deinsum_op_impl.h580 TF_RETURN_IF_ERROR(ReshapeToRank3(inputs[1], bcast.y_batch_size(), &rhs)); in ContractOperands()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dunroll_batch_matmul.cc205 SliceInput(input_rhs, base_name, "b", bcast.y_batch_size(), input_array_b, in Run()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dunroll_batch_matmul.cc293 sliceInput(input_rhs, bcast.y_batch_size(), loc, rewriter); in matchAndRewrite()
/external/tensorflow/tensorflow/python/keras/
Dbackend.py1764 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: