Searched refs:ValidateResultShape (Results 1 – 3 of 3) sorted by relevance
160 TF_RETURN_IF_ERROR(ValidateResultShape(*build_options.result_layout(), in CompileExecutable()
269 Status ValidateResultShape(const Shape& client_shape,
216 Status Service::ValidateResultShape(const Shape& client_shape, in ValidateResultShape() function in xla::Service296 ValidateResultShape(shape_with_output_layout, program_shape.result())); in CreateModuleConfig()