Searched refs:ValidateResultShape (Results 1 – 5 of 5) sorted by relevance
30 Status ValidateResultShape(const Shape& client_shape, in ValidateResultShape() function78 ValidateResultShape(shape_with_output_layout, program_shape.result())); in CreateModuleConfig()
142 TF_RETURN_IF_ERROR(ValidateResultShape(*build_options.result_layout(), in CompileExecutables()
271 Status ValidateResultShape(const Shape& client_shape,
218 Status Service::ValidateResultShape(const Shape& client_shape, in ValidateResultShape() function in xla::Service
42 Status ValidateResultShape(const Shape& client_shape, in ValidateResultShape() function90 ValidateResultShape(result_layout.value(), program_shape.result())); in CreateModuleConfig()