Home
last modified time | relevance | path

Searched refs:DynamicShapeHandlingMode (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/
H A Dhlo_function_importer.h47 enum class DynamicShapeHandlingMode { kDynamic, kConvertToStatic }; enum
78 DynamicShapeHandlingMode mode = DynamicShapeHandlingMode::kDynamic);
173 DynamicShapeHandlingMode mode = DynamicShapeHandlingMode::kDynamic);
179 DynamicShapeHandlingMode mode = DynamicShapeHandlingMode::kDynamic);
H A Dhlo_function_importer.cc470 mlir::OpBuilder* builder, DynamicShapeHandlingMode mode) { in ImportInstruction()
485 mlir::OpBuilder* func_builder, DynamicShapeHandlingMode mode) { in ImportInstructionImpl()
487 const Shape& shape = mode == DynamicShapeHandlingMode::kConvertToStatic in ImportInstructionImpl()
1550 mlir::OpBuilder* func_builder, DynamicShapeHandlingMode mode) { in ImportInstructionWithLayout()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dmhlo_to_lhlo_with_xla.cc317 instr, loads, &b, xla::DynamicShapeHandlingMode::kConvertToStatic)); in CreateOpInFusion()