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

  /external/tensorflow/tensorflow/core/util/
bcast.h 118 static Eigen::array<Eigen::DenseIndex, NDIMS> ToIndexArray(
  /external/tensorflow/tensorflow/core/kernels/
substr_op.cc 156 input.broadcast(BCast::ToIndexArray<1>(bcast.x_bcast()));
166 pos_shaped.broadcast(BCast::ToIndexArray<1>(bcast.y_bcast()));
176 len_shaped.broadcast(BCast::ToIndexArray<1>(bcast.y_bcast()));
219 input.broadcast(BCast::ToIndexArray<2>(bcast.x_bcast()));
229 pos_shaped.broadcast(BCast::ToIndexArray<2>(bcast.y_bcast()));
239 len_shaped.broadcast(BCast::ToIndexArray<2>(bcast.y_bcast()));
xent_op.cc 90 BCast::ToIndexArray<2>(bcast.x_bcast()),
91 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 127 BCast::ToIndexArray<2>(bcast->x_bcast()),
129 BCast::ToIndexArray<2>(bcast->y_bcast()), error_ptr);
134 BCast::ToIndexArray<3>(bcast->x_bcast()),
136 BCast::ToIndexArray<3>(bcast->y_bcast()), error_ptr);
141 BCast::ToIndexArray<4>(bcast->x_bcast()),
143 BCast::ToIndexArray<4>(bcast->y_bcast()), error_ptr);
148 BCast::ToIndexArray<5>(bcast->x_bcast()),
150 BCast::ToIndexArray<5>(bcast->y_bcast()), error_ptr);
sparse_dense_binary_op_shared.cc 126 .broadcast(BCast::ToIndexArray<NDIM>(b.y_bcast())); \

Completed in 415 milliseconds