Home
last modified time | relevance | path

Searched refs:TensorFlowDeviceDialect (Results 1 – 18 of 18) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
H A Dtpu_rewrite_device_util_test.cc630 context.loadDialect<mlir::tf_device::TensorFlowDeviceDialect>(); in TEST()
648 context.loadDialect<mlir::tf_device::TensorFlowDeviceDialect>(); in TEST()
667 context.loadDialect<mlir::tf_device::TensorFlowDeviceDialect>(); in TEST()
686 context.loadDialect<mlir::tf_device::TensorFlowDeviceDialect>(); in TEST()
703 context.loadDialect<mlir::tf_device::TensorFlowDeviceDialect>(); in TEST()
721 context.loadDialect<mlir::tf_device::TensorFlowDeviceDialect>(); in TEST()
741 context.loadDialect<mlir::tf_device::TensorFlowDeviceDialect>(); in TEST()
761 context.loadDialect<mlir::tf_device::TensorFlowDeviceDialect>(); in TEST()
784 context.loadDialect<mlir::tf_device::TensorFlowDeviceDialect>(); in TEST()
810 context.loadDialect<mlir::tf_device::TensorFlowDeviceDialect>(); in TEST()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
H A Dtf_device.h40 class TensorFlowDeviceDialect : public Dialect {
44 explicit TensorFlowDeviceDialect(MLIRContext* context);
H A Dtf_device.cc119 TensorFlowDeviceDialect::TensorFlowDeviceDialect(MLIRContext* context) in TensorFlowDeviceDialect() function in mlir::tf_device::TensorFlowDeviceDialect
121 TypeID::get<TensorFlowDeviceDialect>()) { in TensorFlowDeviceDialect()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dpasses_detail.h33 class TensorFlowDeviceDialect; variable
H A Dtf_device_passes.td282 let dependentDialects = ["tf_device::TensorFlowDeviceDialect"];
288 let dependentDialects = ["tf_device::TensorFlowDeviceDialect"];
H A Ddecompose_reduce_dataset.cc62 registry.insert<tf_device::TensorFlowDeviceDialect>(); in getDependentDialects()
H A Dtpu_cluster_formation.cc82 registry.insert<tf_device::TensorFlowDeviceDialect>(); in getDependentDialects()
H A Dtf_passes.td300 let dependentDialects = ["tf_device::TensorFlowDeviceDialect"];
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/
H A Ddialect_registration.h39 mlir::tf_device::TensorFlowDeviceDialect, in RegisterAllTensorFlowDialects()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/
H A Dtf_cpurt_test_passes.td30 let dependentDialects = ["mlir::tf_device::TensorFlowDeviceDialect"];
H A Dtf_cpurt_passes.td88 let dependentDialects = ["mlir::tf_device::TensorFlowDeviceDialect"];
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/transforms/
H A Dtf_jitrt_test_passes.td30 let dependentDialects = ["mlir::tf_device::TensorFlowDeviceDialect"];
H A Dtf_jitrt_passes.td198 let dependentDialects = ["mlir::tf_device::TensorFlowDeviceDialect"];
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dop_to_device_cluster.cc106 registry.insert<mlir::tf_device::TensorFlowDeviceDialect>(); in getDependentDialects()
H A Dtpu_integration.cc133 registry.insert<mlir::tf_device::TensorFlowDeviceDialect>(); in getDependentDialects()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfr/integration/
H A Dtfr_decompose_ctx.cc102 mlir::tf_device::TensorFlowDeviceDialect, in GetFromText()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/
H A Dmlir_graph_optimization_pass.cc132 mlir::tf_device::TensorFlowDeviceDialect, in RegisterDialects()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/
H A Dtf_to_tfrt.cc1500 target->addIllegalDialect<tf_device::TensorFlowDeviceDialect>(); in SetUpTFToTFRTConversionLegality()