Searched refs:IdentityShapeRepresentationFn (Results 1 – 6 of 6) sorted by relevance
287 IdentityShapeRepresentationFn(); in CompileMlirToXlaHloViaBuilder()317 IdentityShapeRepresentationFn(), in MlirTfToHloTextTranslateFunctionImpl()345 emit_use_tuple_arg, emit_return_tuple, IdentityShapeRepresentationFn(), in MlirTfGraphToHloTextTranslateFunction()
416 *shape_representation_fn = IdentityShapeRepresentationFn(); in CompileMlirSetup()
83 XlaHelpers::ShapeRepresentationFn IdentityShapeRepresentationFn();
129 XlaHelpers::ShapeRepresentationFn IdentityShapeRepresentationFn() { in IdentityShapeRepresentationFn() function
527 options_.shape_representation_fn = IdentityShapeRepresentationFn(); in XlaCompiler()
455 shape_representation_fn_ = tensorflow::IdentityShapeRepresentationFn(); in ConvertToHloModule()