| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| cwise_ops.cc | 96 // rhs.reshape(bcast.y_reshape()).broadcast(bcast.y_bcast()) have
112 builder->Broadcast(rhs_shaped, broadcast_helper.y_bcast());
113 int rhs_size = broadcast_helper.y_bcast().size();
|
| /external/tensorflow/tensorflow/core/kernels/ |
| substr_op.cc | 131 pos_shaped.broadcast(BCast::ToIndexArray<1>(bcast.y_bcast()));
141 len_shaped.broadcast(BCast::ToIndexArray<1>(bcast.y_bcast()));
180 pos_shaped.broadcast(BCast::ToIndexArray<2>(bcast.y_bcast()));
190 len_shaped.broadcast(BCast::ToIndexArray<2>(bcast.y_bcast()));
|
| sparse_dense_binary_op_shared.cc | 121 // "b.y_reshape()" and "b.y_bcast()" are guaranteed to have rank "ndims".
129 .broadcast(BCast::ToIndexArray<NDIM>(b.y_bcast())); \
|
| cwise_ops_common.h | 127 BCast::ToIndexArray<2>(bcast->y_bcast()), error_ptr);
134 BCast::ToIndexArray<3>(bcast->y_bcast()), error_ptr);
141 BCast::ToIndexArray<4>(bcast->y_bcast()), error_ptr);
148 BCast::ToIndexArray<5>(bcast->y_bcast()), error_ptr);
|
| quantized_mul_op.cc | 362 LOG(INFO) << "bcast.y_bcast()="
363 << TensorShape(bcast.y_bcast()).DebugString();
|
| quantized_add_op.cc | 554 LOG(INFO) << "bcast.y_bcast()="
555 << TensorShape(bcast.y_bcast()).DebugString();
|
| /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/compiler/xla/tests/ |
| concat_test.cc | 580 auto y_bcast = builder.Broadcast(y, {1, 5, 7}); local
582 auto concat = builder.ConcatInDim({y_bcast, x}, /*dimension=*/0);
|
