Searched refs:dims_handle (Results 1 – 1 of 1) sorted by relevance
320 std::vector<shape_inference::DimensionHandle> dims_handle; in InferenceShape() local326 dims_handle.push_back(inference_context->MakeDim(dims)); in InferenceShape()328 *shape_handle = inference_context->MakeShape(dims_handle); in InferenceShape()335 std::vector<shape_inference::DimensionHandle> dims_handle( in InferenceShape() local337 for (int i = 0; i < dims_handle.size(); i++) { in InferenceShape()338 dims_handle[i] = inference_context->MakeDim(inference_shape_.dim_size(i)); in InferenceShape()340 *shape_handle = inference_context->MakeShape(dims_handle); in InferenceShape()