Searched defs:tensor_idx (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dsparse_conditional_accumulator.h87 const Tensor* tensor_idx = std::get<0>(*tensor); local
112 tensor_idx->dims() > 0) {
113 for (int64 i = 0; i < tensor_idx->dim_size(0); i++) {
114 if (tensor_idx->vec<int64>()(i) >= shape_.dim_size(0)) {
117 "; index is ", tensor_idx->vec<int64>()(i), " exceeded ",

Completed in 387 milliseconds