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

  /external/tensorflow/tensorflow/core/kernels/
sparse_reorder_op.cc 39 const Tensor& input_ind = context->input(0); variable
40 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_ind.shape()),
43 input_ind.shape().DebugString()));
63 sparse::SparseTensor input_sp(input_ind, input_val, input_shape, std_order);
70 sparse::SparseTensor reordered_sp(tensor::DeepCopy(input_ind),
reshape_util.cc 126 auto input_ind = input_indices_in.matrix<int64>(); local
131 id += input_ind(i, j) * input_strides[j];

Completed in 73 milliseconds