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

  /external/tensorflow/tensorflow/core/kernels/
sparse_conditional_accumulator.h 87 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 ",
depthwise_conv_op_gpu.cu.cc 227 const int tensor_idx = thread_pix * in_depth + thread_depth;
251 const int filter_offset = tensor_idx + start_channel;
506 const int tensor_idx = thread_depth * in_pixels + thread_pix;
532 const int inout_offset = channel * in_pixels + tensor_idx;
    [all...]

Completed in 62 milliseconds