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

  /external/tensorflow/tensorflow/core/kernels/
dynamic_stitch_op_gpu.cu.cc 37 int32* data_indices = GetCudaDeviceArrayOnDevice(&input_indices); local
42 const int32 input_index = data_indices[slice_id];
dynamic_partition_op.cc 152 Eigen::DSizes<Eigen::DenseIndex, 2> data_indices(i, 0);
154 data_flat.slice(data_indices, sizes);
dynamic_stitch_op.cc 264 Eigen::DSizes<Eigen::DenseIndex, 2> data_indices(i, 0);
271 data_flat.slice(data_indices, sizes);

Completed in 65 milliseconds