Searched refs:TensorFlowDeviceDialect (Results 1 – 18 of 18) sorted by relevance
40 class TensorFlowDeviceDialect : public Dialect {44 explicit TensorFlowDeviceDialect(MLIRContext* context);
119 TensorFlowDeviceDialect::TensorFlowDeviceDialect(MLIRContext* context) in TensorFlowDeviceDialect() function in mlir::tf_device::TensorFlowDeviceDialect121 TypeID::get<TensorFlowDeviceDialect>()) { in TensorFlowDeviceDialect()
630 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 …]
33 class TensorFlowDeviceDialect; variable
282 let dependentDialects = ["tf_device::TensorFlowDeviceDialect"];288 let dependentDialects = ["tf_device::TensorFlowDeviceDialect"];
62 registry.insert<tf_device::TensorFlowDeviceDialect>(); in getDependentDialects()
82 registry.insert<tf_device::TensorFlowDeviceDialect>(); in getDependentDialects()
300 let dependentDialects = ["tf_device::TensorFlowDeviceDialect"];
39 mlir::tf_device::TensorFlowDeviceDialect, in RegisterAllTensorFlowDialects()
30 let dependentDialects = ["mlir::tf_device::TensorFlowDeviceDialect"];
88 let dependentDialects = ["mlir::tf_device::TensorFlowDeviceDialect"];
198 let dependentDialects = ["mlir::tf_device::TensorFlowDeviceDialect"];
106 registry.insert<mlir::tf_device::TensorFlowDeviceDialect>(); in getDependentDialects()
133 registry.insert<mlir::tf_device::TensorFlowDeviceDialect>(); in getDependentDialects()
102 mlir::tf_device::TensorFlowDeviceDialect, in GetFromText()
132 mlir::tf_device::TensorFlowDeviceDialect, in RegisterDialects()
1500 target->addIllegalDialect<tf_device::TensorFlowDeviceDialect>(); in SetUpTFToTFRTConversionLegality()