Home
last modified time | relevance | path

Searched refs:y_bcast (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsubstr_op.cc162 pos_shaped.broadcast(BCast::ToIndexArray<1>(bcast.y_bcast())); in Compute()
172 len_shaped.broadcast(BCast::ToIndexArray<1>(bcast.y_bcast())); in Compute()
215 pos_shaped.broadcast(BCast::ToIndexArray<2>(bcast.y_bcast())); in Compute()
225 len_shaped.broadcast(BCast::ToIndexArray<2>(bcast.y_bcast())); in Compute()
Dcwise_op_select.cc202 BCast::ToIndexArray<NDIMS>(cond_bcast.y_bcast()), \ in Compute()
203 BCast::ToIndexArray<NDIMS>(then_bcast.y_bcast()), \ in Compute()
204 BCast::ToIndexArray<NDIMS>(else_bcast.y_bcast())); \ in Compute()
Dxent_op.cc88 BCast::ToIndexArray<2>(bcast.y_bcast()), in Compute()
Dsparse_dense_binary_op_shared.cc126 .broadcast(BCast::ToIndexArray<NDIM>(b.y_bcast())); \ in Compute()
Dcwise_ops_common.h190 BCast::ToIndexArray<2>(bcast.y_bcast()), error_ptr); in Compute()
197 BCast::ToIndexArray<3>(bcast.y_bcast()), error_ptr); in Compute()
204 BCast::ToIndexArray<4>(bcast.y_bcast()), error_ptr); in Compute()
211 BCast::ToIndexArray<5>(bcast.y_bcast()), error_ptr); in Compute()
Dquantized_mul_op.cc362 << TensorShape(bcast.y_bcast()).DebugString(); in Compute()
Dquantized_add_op.cc554 << TensorShape(bcast.y_bcast()).DebugString(); in Compute()
/external/tensorflow/tensorflow/core/util/
Dmatmul_bcast.h60 batch_bcast_->y_bcast(), &y_batch_indices_); in MatMulBCast()
Dbcast_test.cc36 strings::StrAppend(&ret, "[", absl::StrJoin(b.y_bcast(), ","), "]"); in BCast()
Dbcast.h377 const Vec& y_bcast() const { return bcast_[1]; } in y_bcast() function
/external/tensorflow/tensorflow/compiler/xla/tests/
Dconcat_test.cc870 auto y_bcast = Broadcast(y, {1, 5, 7}); in XLA_TEST_F() local
872 auto concat = ConcatInDim(&builder, {y_bcast, x}, /*dimension=*/0); in XLA_TEST_F()