HomeSort by relevance Sort by last modified time
    Searched refs:ToIndexArray (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
substr_op.cc 121 input.broadcast(BCast::ToIndexArray<1>(bcast.x_bcast()));
131 pos_shaped.broadcast(BCast::ToIndexArray<1>(bcast.y_bcast()));
141 len_shaped.broadcast(BCast::ToIndexArray<1>(bcast.y_bcast()));
170 input.broadcast(BCast::ToIndexArray<2>(bcast.x_bcast()));
180 pos_shaped.broadcast(BCast::ToIndexArray<2>(bcast.y_bcast()));
190 len_shaped.broadcast(BCast::ToIndexArray<2>(bcast.y_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()), \
cwise_ops_common.h 125 BCast::ToIndexArray<2>(bcast->x_bcast()),
127 BCast::ToIndexArray<2>(bcast->y_bcast()), error_ptr);
132 BCast::ToIndexArray<3>(bcast->x_bcast()),
134 BCast::ToIndexArray<3>(bcast->y_bcast()), error_ptr);
139 BCast::ToIndexArray<4>(bcast->x_bcast()),
141 BCast::ToIndexArray<4>(bcast->y_bcast()), error_ptr);
146 BCast::ToIndexArray<5>(bcast->x_bcast()),
148 BCast::ToIndexArray<5>(bcast->y_bcast()), error_ptr);
sparse_dense_binary_op_shared.cc 129 .broadcast(BCast::ToIndexArray<NDIM>(b.y_bcast())); \
  /external/tensorflow/tensorflow/core/util/
bcast.h 109 static Eigen::array<Eigen::DenseIndex, NDIMS> ToIndexArray(

Completed in 264 milliseconds