Home
last modified time | relevance | path

Searched refs:IsBroadcastingRequired (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dmatmul_bcast_test.cc41 EXPECT_TRUE(bcast.IsBroadcastingRequired()); in TEST()
54 EXPECT_FALSE(bcast.IsBroadcastingRequired()); in TEST()
67 EXPECT_FALSE(bcast.IsBroadcastingRequired()); in TEST()
80 EXPECT_TRUE(bcast1.IsBroadcastingRequired()); in TEST()
89 EXPECT_TRUE(bcast2.IsBroadcastingRequired()); in TEST()
Dmatmul_bcast.h37 bool IsBroadcastingRequired() const { return broadcasting_required_; } in IsBroadcastingRequired() function
Dbcast.h89 bool IsBroadcastingRequired() const { return broadcasting_required_; } in IsBroadcastingRequired() function
/external/tensorflow/tensorflow/core/kernels/
Dbatch_matmul_op_impl.h96 const bool should_bcast = bcast.IsBroadcastingRequired(); in Run()
133 const bool should_bcast = bcast.IsBroadcastingRequired();
174 const bool should_bcast = bcast.IsBroadcastingRequired();
334 if (!bcast.IsBroadcastingRequired()) {
470 if (!bcast.IsBroadcastingRequired()) {
557 const bool should_bcast = bcast.IsBroadcastingRequired();
Dmatrix_triangular_solve_op_impl.h87 const bool should_bcast = bcast.IsBroadcastingRequired(); in Run()
266 if (!bcast.IsBroadcastingRequired() || out->shape() == in_y.shape()) {
357 if (!bcast.IsBroadcastingRequired()) {
Dmkl_batch_matmul_op.cc157 if (!bcast.IsBroadcastingRequired()) { in Compute()
Drandom_binomial_op.cc189 const bool should_bcast = bcast.IsBroadcastingRequired(); in operator ()()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dunroll_batch_matmul.cc214 const int a_batch_idx = bcast.IsBroadcastingRequired() in Run()
217 const int b_batch_idx = bcast.IsBroadcastingRequired() in Run()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dunroll_batch_matmul.cc169 if (bcast.IsBroadcastingRequired()) { in createMatMulOps()