Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/util/
Dmatmul_bcast_test.cc43 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()
117 EXPECT_EQ(3, bcast.x_batch_size()); in TEST()
129 EXPECT_EQ(3, bcast.x_batch_size()); in TEST()
Dmatmul_bcast.h40 const int64 x_batch_size() const { return x_batch_size_; } in x_batch_size() function
/external/tensorflow/tensorflow/core/kernels/
Dmatrix_triangular_solve_op_impl.h180 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) {
Dbatch_matmul_op_impl.h324 a_device_memory.reserve(bcast.x_batch_size());
346 for (int64 i = 0; i < bcast.x_batch_size(); ++i) {
460 a_device_memory.reserve(bcast.x_batch_size());
482 for (int64 i = 0; i < bcast.x_batch_size(); ++i) {
616 in0_reshaped.CopyFrom(in0, TensorShape({bcast.x_batch_size(), d0, d1})),
Deinsum_op_impl.h578 TF_RETURN_IF_ERROR(ReshapeToRank3(inputs[0], bcast.x_batch_size(), &lhs)); in ContractOperands()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dunroll_batch_matmul.cc202 SliceInput(input_lhs, base_name, "a", bcast.x_batch_size(), input_array_a, in Run()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dunroll_batch_matmul.cc291 sliceInput(input_lhs, bcast.x_batch_size(), loc, rewriter); in matchAndRewrite()
/external/tensorflow/tensorflow/python/keras/
Dbackend.py1763 x_batch_size = x_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: