Home
last modified time | relevance | path

Searched refs:ToIndexArray (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsubstr_op.cc156 input.broadcast(BCast::ToIndexArray<1>(bcast.x_bcast())); in Compute()
166 pos_shaped.broadcast(BCast::ToIndexArray<1>(bcast.y_bcast())); in Compute()
176 len_shaped.broadcast(BCast::ToIndexArray<1>(bcast.y_bcast())); in Compute()
219 input.broadcast(BCast::ToIndexArray<2>(bcast.x_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()
Dbetainc_op.cc98 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()
Dcwise_ops_common.h127 BCast::ToIndexArray<2>(bcast->x_bcast()), in Compute()
129 BCast::ToIndexArray<2>(bcast->y_bcast()), error_ptr); in Compute()
134 BCast::ToIndexArray<3>(bcast->x_bcast()), in Compute()
136 BCast::ToIndexArray<3>(bcast->y_bcast()), error_ptr); in Compute()
141 BCast::ToIndexArray<4>(bcast->x_bcast()), in Compute()
143 BCast::ToIndexArray<4>(bcast->y_bcast()), error_ptr); in Compute()
148 BCast::ToIndexArray<5>(bcast->x_bcast()), in Compute()
150 BCast::ToIndexArray<5>(bcast->y_bcast()), error_ptr); in Compute()
Dxent_op.cc90 BCast::ToIndexArray<2>(bcast.x_bcast()), in Compute()
91 BCast::ToIndexArray<2>(bcast.y_bcast()), in Compute()
Dsparse_dense_binary_op_shared.cc126 .broadcast(BCast::ToIndexArray<NDIM>(b.y_bcast())); \ in Compute()
/external/tensorflow/tensorflow/core/util/
Dbcast.h118 static Eigen::array<Eigen::DenseIndex, NDIMS> ToIndexArray( in ToIndexArray() function