| /external/tensorflow/tensorflow/core/util/ |
| bcast.h | 52 // y.reshape(b.y_reshape()).broadcast(b.y_bcast())
98 const Vec& y_bcast() const { return y_bcast_; } function in class:tensorflow::BCast
|
| bcast_test.cc | 36 strings::StrAppend(&ret, "[", str_util::Join(b.y_bcast(), ","), "]");
182 "[1,1,1,1,1,1,1,1,1][11,7,5,1,1,3,2,1,1]" // y_reshape(), y_bcast()
196 "[11,7,5,1,1,3,2,1,1][1,1,1,1,1,1,1,1,1]" // y_reshape(), y_bcast()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| substr_op.cc | 166 pos_shaped.broadcast(BCast::ToIndexArray<1>(bcast.y_bcast()));
176 len_shaped.broadcast(BCast::ToIndexArray<1>(bcast.y_bcast()));
229 pos_shaped.broadcast(BCast::ToIndexArray<2>(bcast.y_bcast()));
239 len_shaped.broadcast(BCast::ToIndexArray<2>(bcast.y_bcast()));
|
| sparse_dense_binary_op_shared.cc | 118 // "b.y_reshape()" and "b.y_bcast()" are guaranteed to have rank "ndims".
126 .broadcast(BCast::ToIndexArray<NDIM>(b.y_bcast())); \
|
| xent_op.cc | 91 BCast::ToIndexArray<2>(bcast.y_bcast()),
|
| cwise_ops_common.h | 129 BCast::ToIndexArray<2>(bcast->y_bcast()), error_ptr);
136 BCast::ToIndexArray<3>(bcast->y_bcast()), error_ptr);
143 BCast::ToIndexArray<4>(bcast->y_bcast()), error_ptr);
150 BCast::ToIndexArray<5>(bcast->y_bcast()), error_ptr);
|
| quantized_mul_op.cc | 361 LOG(INFO) << "bcast.y_bcast()="
362 << TensorShape(bcast.y_bcast()).DebugString();
|
| quantized_add_op.cc | 553 LOG(INFO) << "bcast.y_bcast()="
554 << TensorShape(bcast.y_bcast()).DebugString();
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| concat_test.cc | 624 auto y_bcast = Broadcast(y, {1, 5, 7}); local
626 auto concat = ConcatInDim(&builder, {y_bcast, x}, /*dimension=*/0);
|
