Searched refs:output_shape_idx (Results 1 – 2 of 2) sorted by relevance
36 int output_indices_idx, int output_shape_idx) { in Reshape() argument105 context->set_output(output_shape_idx, input_shape_in); in Reshape()144 OP_REQUIRES_OK(context, context->allocate_output(output_shape_idx, in Reshape()
27 int output_indices_idx, int output_shape_idx);