Home
last modified time | relevance | path

Searched defs:shape_representation_fn (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/
H A Dlayout_util.cc24 const ShapeRepresentationFn& shape_representation_fn, in RewriteLayoutWithShardedShape()
69 const ShapeRepresentationFn& shape_representation_fn, in ReshapeWithCorrectRepresentationAndSharding()
H A Dmlir_hlo_to_hlo.h48 ShapeRepresentationFn shape_representation_fn; member
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dlayout_util.h55 XlaHelpers::ShapeRepresentationFn shape_representation_fn; member
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_module_util.cc150 HloModule* module, DeviceShapeRepresentationFn shape_representation_fn, in UpdateEntryComputationLayout()
H A Dtransfer_manager.cc383 int device_ordinal, DeviceShapeRepresentationFn shape_representation_fn) { in AllocateScopedShapedBuffer()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
H A Dtf_xla_mlir_translate.cc300 XlaHelpers::ShapeRepresentationFn shape_representation_fn = in CompileMlirToXlaHloViaBuilder() local