Searched refs:TensorFlowDialect (Results 1 – 25 of 31) sorted by relevance
12
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops.cc | 90 return TensorFlowDialect::constantFold(op, operands, results); in fold()
98 return TensorFlowDialect::decode(input, output); in decode()
137 return !wouldBeCloned || TensorFlowDialect::CanDuplicate(op); in isLegalToInline()
165 bool TensorFlowDialect::CanDuplicate(Operation *op) { in CanDuplicate()
184 bool TensorFlowDialect::CanHaveSideEffects(Operation *op) { in CanHaveSideEffects()
200 std::vector<TensorFlowDialect::AdditionalOpFunction>
201 *TensorFlowDialect::GetAdditionalOperationHooks() { in GetAdditionalOperationHooks()
203 new std::vector<TensorFlowDialect::AdditionalOpFunction>(); in GetAdditionalOperationHooks()
207 TensorFlowDialect::ConstantFoldHook TensorFlowDialect::constant_fold_hook_;
208 TensorFlowDialect::DecodeConstantHook TensorFlowDialect::decode_constant_hook_;
[all …]
|
| D | tf_ops.h | 51 class TensorFlowDialect : public Dialect {
53 TensorFlowDialect(MLIRContext *context);
99 typedef std::function<void(TensorFlowDialect &dialect)> AdditionalOpFunction;
|
| D | tf_saved_model.cc | 129 context->loadDialect<TF::TensorFlowDialect>(); in TensorFlowSavedModelDialect()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ |
| D | dialect_registration.h | 31 registry.insert<mlir::StandardOpsDialect, mlir::TF::TensorFlowDialect, in RegisterAllTensorFlowDialects()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | decode_attributes_hook.cc | 51 TF::TensorFlowDialect::RegisterDecodeConstantHook(DecodeOpaqueTensorHook); in __anon761463770202()
|
| D | tf_device_assignment.cc | 37 Dialect* tf = getContext().getLoadedDialect<TensorFlowDialect>(); in runOnFunction()
|
| D | tf_device_replication_pass.cc | 36 registry.insert<TF::TensorFlowDialect>(); in getDependentDialects()
|
| D | constant_fold.cc | 165 TensorFlowDialect::RegisterConstantFoldHook(ConstantFoldFallbackHook); in __anon3cd6eecb0502()
|
| D | legalize_hlo.cc | 948 registry.insert<TF::TensorFlowDialect>(); in getDependentDialects()
1237 target.addLegalDialect<TensorFlowDialect>(); in runOnFunction()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfjs/ |
| D | tfjs_opt.cc | 30 registry.insert<mlir::TF::TensorFlowDialect>(); in main()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/passes/ |
| D | tfr_opt.cc | 33 registry.insert<mlir::scf::SCFDialect, mlir::TF::TensorFlowDialect, in main()
|
| D | canonicalize.cc | 157 mlir::Dialect *tf = context->getLoadedDialect<mlir::TF::TensorFlowDialect>(); in populateCanonicalizationPatterns()
|
| D | raise_to_tf.cc | 452 registry.insert<TFRDialect, TF::TensorFlowDialect, scf::SCFDialect, in getDependentDialects()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/python/ |
| D | tfr_wrapper.cc | 37 registry.insert<mlir::scf::SCFDialect, mlir::TF::TensorFlowDialect, in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/analysis/ |
| D | side_effect_analysis.cc | 230 TF::TensorFlowDialect::getDialectNamespace()) in OpIsKnownToHaveNoSideEffect()
231 return !TensorFlowDialect::CanHaveSideEffects(op); in OpIsKnownToHaveNoSideEffect()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_translate.cc | 187 registry.insert<TF::TensorFlowDialect>(); in __anonacaeaea80302()
|
| D | mlir_tflite_runner.cc | 106 registry.insert<mlir::TF::TensorFlowDialect, mlir::TFL::TensorFlowLiteDialect, in main()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | convert_tensor_test.cc | 38 context.loadDialect<mlir::TF::TensorFlowDialect>(); in RegisterDialects()
|
| D | shape_inference_utils.cc | 181 TensorFlowDialect::getDialectNamespace()); in InferReturnTypeComponentsForTFOp()
|
| D | tf_xla_mlir_translate.cc | 357 registry.insert<mlir::StandardOpsDialect, mlir::TF::TensorFlowDialect>(); in RegisterMlirInputDialects()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | perception_ops_utils_test.cc | 109 context_->loadDialect<mlir::StandardOpsDialect, mlir::TF::TensorFlowDialect, in SetUp()
|
| D | lstm_utils_test.cc | 97 context_->loadDialect<mlir::StandardOpsDialect, mlir::TF::TensorFlowDialect, in SetUp()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/integration/ |
| D | tfr_decompose_ctx.cc | 100 mlir::TF::TensorFlowDialect, in GetFromText()
|
| /external/tensorflow/tensorflow/compiler/mlir/ |
| D | mlir_graph_optimization_pass.cc | 112 mlir::TF::TensorFlowDialect, in RegisterDialects()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | export_graphdef.cc | 649 auto grad_string = mlir::TF::TensorFlowDialect::GetGradientAttrName(); in ConvertLibFunction()
663 auto stateful_string = mlir::TF::TensorFlowDialect::GetStatefulAttrName(); in ConvertLibFunction()
|
12