Searched refs:x_bcast (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | betainc_op.cc | 98 BCast::ToIndexArray<NDIM>(a_shaper.x_bcast()), b_value, \ in Compute()
99 BCast::ToIndexArray<NDIM>(b_shaper.x_bcast()), x_value, \ in Compute()
100 BCast::ToIndexArray<NDIM>(x_shaper.x_bcast()), \ in Compute()
|
| D | broadcast_to_op.h | 59 BCast::ToIndexArrayType<int, NDIMS>(bcast.x_bcast())); in ReshapeAndBCast()
64 BCast::ToIndexArrayType<Eigen::DenseIndex, NDIMS>(bcast.x_bcast())); in ReshapeAndBCast()
|
| D | substr_op.cc | 156 input.broadcast(BCast::ToIndexArray<1>(bcast.x_bcast())); in Compute()
219 input.broadcast(BCast::ToIndexArray<2>(bcast.x_bcast())); in Compute()
|
| D | cwise_ops_common.h | 127 BCast::ToIndexArray<2>(bcast->x_bcast()), in Compute()
134 BCast::ToIndexArray<3>(bcast->x_bcast()), in Compute()
141 BCast::ToIndexArray<4>(bcast->x_bcast()), in Compute()
148 BCast::ToIndexArray<5>(bcast->x_bcast()), in Compute()
|
| D | xent_op.cc | 90 BCast::ToIndexArray<2>(bcast.x_bcast()), in Compute()
|
| D | quantized_mul_op.cc | 360 << TensorShape(bcast.x_bcast()).DebugString(); in Compute()
|
| D | quantized_add_op.cc | 552 << TensorShape(bcast.x_bcast()).DebugString(); in Compute()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | bcast.h | 96 const Vec& x_bcast() const { return x_bcast_; } in x_bcast() function
|
| D | bcast_test.cc | 34 strings::StrAppend(&ret, "[", str_util::Join(b.x_bcast(), ","), "]"); in BCast()
|