Searched defs:shape_representation_fn (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | compile_mlir_util.cc | 85 const XlaHelpers::ShapeRepresentationFn shape_representation_fn, in GetXlaInputShapes()
138 const XlaHelpers::ShapeRepresentationFn shape_representation_fn, in GetOutputInfo()
392 const XlaHelpers::ShapeRepresentationFn shape_representation_fn, in ConvertMLIRToXlaComputation()
408 XlaHelpers::ShapeRepresentationFn* shape_representation_fn) { in CompileMlirSetup()
431 XlaHelpers::ShapeRepresentationFn shape_representation_fn; in BuildHloFromTf() local
449 XlaHelpers::ShapeRepresentationFn shape_representation_fn, in PopulateResultIOInfo()
471 XlaHelpers::ShapeRepresentationFn shape_representation_fn, in CompileMlirToXlaHlo()
493 const XlaHelpers::ShapeRepresentationFn shape_representation_fn, in CompileSerializedMlirToXlaHlo()
634 const XlaHelpers::ShapeRepresentationFn shape_representation_fn, in CompileGraphToXlaHlo()
701 const XlaHelpers::ShapeRepresentationFn shape_representation_fn, in CompileGraphToXlaHlo()
|
| D | tf_xla_mlir_translate.cc | 286 XlaHelpers::ShapeRepresentationFn shape_representation_fn = in CompileMlirToXlaHloViaBuilder() local
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device.h | 72 const XlaCompiler::ShapeRepresentationFn& shape_representation_fn() const { in shape_representation_fn() function
128 XlaCompiler::ShapeRepresentationFn shape_representation_fn; member
|
| D | xla_device_context.h | 82 const XlaCompiler::ShapeRepresentationFn& shape_representation_fn() const { in shape_representation_fn() function
|
| D | xla_device_context.cc | 82 XlaCompiler::ShapeRepresentationFn shape_representation_fn, in XlaDeviceContext()
|
| D | xla_device.cc | 148 XlaCompiler::ShapeRepresentationFn shape_representation_fn, in Metadata()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_helpers.cc | 141 XlaHelpers::ShapeRepresentationFn shape_representation_fn, in RewriteLayoutWithShardedShape()
182 XlaHelpers::ShapeRepresentationFn shape_representation_fn, in ReshapeWithCorrectRepresentationAndSharding()
|
| D | xla_compiler.h | 176 ShapeRepresentationFn shape_representation_fn; member
|
| D | xla_compiler.cc | 171 const XlaCompiler::ShapeRepresentationFn& shape_representation_fn, in BuildComputation()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | type_to_shape.cc | 84 mlir::Type type, CustomShapeRepresentationFn shape_representation_fn) { in TypeToShape()
|
| D | mlir_hlo_to_hlo.cc | 446 tensorflow::XlaHelpers::ShapeRepresentationFn shape_representation_fn, in ConvertToHloModule()
1769 const tensorflow::XlaHelpers::ShapeRepresentationFn shape_representation_fn, in ConvertMlirHloToHlo()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compile_op_common.cc | 257 const XlaCompiler::ShapeRepresentationFn shape_representation_fn, in GetShardingInfo()
280 const XlaCompiler::ShapeRepresentationFn shape_representation_fn, in CompileTFFunctionToHlo()
|