Searched refs:input_indices_in (Results 1 – 2 of 2) sorted by relevance
34 void Reshape(OpKernelContext *context, const Tensor &input_indices_in, in Reshape() argument37 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(input_indices_in.shape()), in Reshape()40 input_indices_in.shape().DebugString())); in Reshape()54 const int64 nnz = input_indices_in.shape().dim_size(0); in Reshape()104 context->set_output(output_indices_idx, input_indices_in); in Reshape()130 auto input_ind = input_indices_in.matrix<int64>(); in Reshape()
25 void Reshape(OpKernelContext *context, const Tensor &input_indices_in,