| /external/tensorflow/tensorflow/core/kernels/ |
| broadcast_to_op.h | 59 BCast::ToIndexArrayType<int, NDIMS>(bcast.x_bcast()));
64 BCast::ToIndexArrayType<Eigen::DenseIndex, NDIMS>(bcast.x_bcast()));
|
| betainc_op.cc | 98 BCast::ToIndexArray<NDIM>(a_shaper.x_bcast()), b_value, \
99 BCast::ToIndexArray<NDIM>(b_shaper.x_bcast()), x_value, \
100 BCast::ToIndexArray<NDIM>(x_shaper.x_bcast()), \
|
| xent_op.cc | 90 BCast::ToIndexArray<2>(bcast.x_bcast()),
|
| substr_op.cc | 156 input.broadcast(BCast::ToIndexArray<1>(bcast.x_bcast()));
219 input.broadcast(BCast::ToIndexArray<2>(bcast.x_bcast()));
|
| cwise_ops_common.h | 127 BCast::ToIndexArray<2>(bcast->x_bcast()),
134 BCast::ToIndexArray<3>(bcast->x_bcast()),
141 BCast::ToIndexArray<4>(bcast->x_bcast()),
148 BCast::ToIndexArray<5>(bcast->x_bcast()),
|
| quantized_mul_op.cc | 359 LOG(INFO) << "bcast.x_bcast()="
360 << TensorShape(bcast.x_bcast()).DebugString();
|
| quantized_add_op.cc | 551 LOG(INFO) << "bcast.x_bcast()="
552 << TensorShape(bcast.x_bcast()).DebugString();
|
| /external/tensorflow/tensorflow/core/util/ |
| bcast.h | 50 // output = x.reshape(b.x_reshape()).broadcast(b.x_bcast())
96 const Vec& x_bcast() const { return x_bcast_; } function in class:tensorflow::BCast
|
| bcast_test.cc | 34 strings::StrAppend(&ret, "[", str_util::Join(b.x_bcast(), ","), "]");
181 "[11,7,5,1,1,3,2,1,1][1,1,1,1,1,1,1,1,1]" // x_reshape(), x_bcast()
195 "[1,1,1,1,1,1,1,1,1][11,7,5,1,1,3,2,1,1]" // x_reshape(), x_bcast()
|
