Searched refs:input_ind (Results 1 – 2 of 2) sorted by relevance
39 const Tensor& input_ind = context->input(0); in Compute() local40 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_ind.shape()), in Compute()43 input_ind.shape().DebugString())); in Compute()65 context, sparse::SparseTensor::Create(input_ind, input_val, input_shape, in Compute()75 sparse::SparseTensor::Create(tensor::DeepCopy(input_ind), in Compute()
130 auto input_ind = input_indices_in.matrix<int64>(); in Reshape() local135 id += input_ind(i, j) * input_strides[j]; in Reshape()