Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
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()
Dsubstr_op.cc162 pos_shaped.broadcast(BCast::ToIndexArray<1>(bcast.y_bcast())); in Compute()
172 len_shaped.broadcast(BCast::ToIndexArray<1>(bcast.y_bcast())); in Compute()
215 pos_shaped.broadcast(BCast::ToIndexArray<2>(bcast.y_bcast())); in Compute()
225 len_shaped.broadcast(BCast::ToIndexArray<2>(bcast.y_bcast())); in Compute()
Dxent_op.cc87 BCast::ToIndexArray<2>(bcast.x_bcast()), in Compute()
88 BCast::ToIndexArray<2>(bcast.y_bcast()), in Compute()
Dcwise_ops_common.h188 BCast::ToIndexArray<2>(bcast.x_bcast()), in Compute()
190 BCast::ToIndexArray<2>(bcast.y_bcast()), error_ptr); in Compute()
195 BCast::ToIndexArray<3>(bcast.x_bcast()), in Compute()
197 BCast::ToIndexArray<3>(bcast.y_bcast()), error_ptr); in Compute()
202 BCast::ToIndexArray<4>(bcast.x_bcast()), in Compute()
204 BCast::ToIndexArray<4>(bcast.y_bcast()), error_ptr); in Compute()
209 BCast::ToIndexArray<5>(bcast.x_bcast()), in Compute()
211 BCast::ToIndexArray<5>(bcast.y_bcast()), error_ptr); in Compute()
Dcwise_op_select.cc202 BCast::ToIndexArray<NDIMS>(cond_bcast.y_bcast()), \ in Compute()
203 BCast::ToIndexArray<NDIMS>(then_bcast.y_bcast()), \ in Compute()
204 BCast::ToIndexArray<NDIMS>(else_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.h410 static Eigen::array<Eigen::DenseIndex, NDIMS> ToIndexArray( in ToIndexArray() function