Searched refs:TensorFlowDeviceDialect (Results 1 – 8 of 8) sorted by relevance
39 class TensorFlowDeviceDialect : public Dialect {43 explicit TensorFlowDeviceDialect(MLIRContext* context);
111 TensorFlowDeviceDialect::TensorFlowDeviceDialect(MLIRContext* context) in TensorFlowDeviceDialect() function in mlir::tf_device::TensorFlowDeviceDialect113 TypeID::get<TensorFlowDeviceDialect>()) { in TensorFlowDeviceDialect()
629 context.loadDialect<mlir::tf_device::TensorFlowDeviceDialect>(); in TEST()645 context.loadDialect<mlir::tf_device::TensorFlowDeviceDialect>(); in TEST()666 context.loadDialect<mlir::tf_device::TensorFlowDeviceDialect>(); in TEST()686 context.loadDialect<mlir::tf_device::TensorFlowDeviceDialect>(); in TEST()706 context.loadDialect<mlir::tf_device::TensorFlowDeviceDialect>(); in TEST()729 context.loadDialect<mlir::tf_device::TensorFlowDeviceDialect>(); in TEST()755 context.loadDialect<mlir::tf_device::TensorFlowDeviceDialect>(); in TEST()782 context.loadDialect<mlir::tf_device::TensorFlowDeviceDialect>(); in TEST()
33 mlir::tf_device::TensorFlowDeviceDialect, in RegisterAllTensorFlowDialects()
43 registry.insert<tf_device::TensorFlowDeviceDialect>(); in getDependentDialects()
75 registry.insert<tf_device::TensorFlowDeviceDialect>(); in getDependentDialects()
101 mlir::tf_device::TensorFlowDeviceDialect, in GetFromText()
114 mlir::tf_device::TensorFlowDeviceDialect, in RegisterDialects()