Searched refs:y_bcast (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | substr_op.cc | 166 pos_shaped.broadcast(BCast::ToIndexArray<1>(bcast.y_bcast())); in Compute()
176 len_shaped.broadcast(BCast::ToIndexArray<1>(bcast.y_bcast())); in Compute()
229 pos_shaped.broadcast(BCast::ToIndexArray<2>(bcast.y_bcast())); in Compute()
239 len_shaped.broadcast(BCast::ToIndexArray<2>(bcast.y_bcast())); in Compute()
|
| D | cwise_ops_common.h | 129 BCast::ToIndexArray<2>(bcast->y_bcast()), error_ptr); in Compute()
136 BCast::ToIndexArray<3>(bcast->y_bcast()), error_ptr); in Compute()
143 BCast::ToIndexArray<4>(bcast->y_bcast()), error_ptr); in Compute()
150 BCast::ToIndexArray<5>(bcast->y_bcast()), error_ptr); in Compute()
|
| D | sparse_dense_binary_op_shared.cc | 126 .broadcast(BCast::ToIndexArray<NDIM>(b.y_bcast())); \ in Compute()
|
| D | xent_op.cc | 91 BCast::ToIndexArray<2>(bcast.y_bcast()), in Compute()
|
| D | quantized_mul_op.cc | 362 << TensorShape(bcast.y_bcast()).DebugString(); in Compute()
|
| D | quantized_add_op.cc | 554 << TensorShape(bcast.y_bcast()).DebugString(); in Compute()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | bcast.h | 98 const Vec& y_bcast() const { return y_bcast_; } in y_bcast() function
|
| D | bcast_test.cc | 36 strings::StrAppend(&ret, "[", str_util::Join(b.y_bcast(), ","), "]"); in BCast()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | concat_test.cc | 624 auto y_bcast = Broadcast(y, {1, 5, 7}); in XLA_TEST_F() local
626 auto concat = ConcatInDim(&builder, {y_bcast, x}, /*dimension=*/0); in XLA_TEST_F()
|