| /external/llvm-project/mlir/include/mlir/IR/ |
| D | MLIRContext.h | 57 Dialect *getLoadedDialect(StringRef name);
62 T *getLoadedDialect() { in getLoadedDialect() function
63 return static_cast<T *>(getLoadedDialect(T::getDialectNamespace())); in getLoadedDialect()
|
| D | FunctionSupport.h | 486 if (auto *dialect = ctx->getLoadedDialect(dialectNamePair.first)) { in verifyTrait()
502 if (auto *dialect = ctx->getLoadedDialect(dialectNamePair.first)) { in verifyTrait()
|
| /external/llvm-project/mlir/lib/IR/ |
| D | Verifier.cpp | 53 return ctx->getLoadedDialect(dialectNamePair.first); in getDialectForAttribute()
221 if (auto *dialect = ctx->getLoadedDialect(dialectPrefix)) in verifyOperation()
|
| D | MLIRContext.cpp | 444 Dialect *MLIRContext::getLoadedDialect(StringRef name) { in getLoadedDialect() function in MLIRContext
451 Dialect *dialect = getLoadedDialect(name); in getOrLoadDialect()
|
| D | Operation.cpp | 224 return getContext()->getLoadedDialect(getName().getDialect()); in getDialect()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tf_device_assignment.cc | 37 Dialect* tf = getContext().getLoadedDialect<TensorFlowDialect>(); in runOnFunction()
|
| D | launch_to_device_attribute.cc | 122 const Dialect* tf_dialect = getContext().getLoadedDialect("tf"); in runOnFunction()
|
| D | tf_device_replication_pass.cc | 41 const Dialect *tf_dialect = getContext().getLoadedDialect("tf"); in runOnOperation()
|
| D | replicate_to_island.cc | 300 const Dialect* tf_dialect = getContext().getLoadedDialect("tf"); in runOnFunction()
|
| D | mark_ops_for_outside_compilation.cc | 323 const Dialect* tf_dialect = getContext().getLoadedDialect("tf"); in runOnOperation()
|
| D | shape_inference.cc | 619 : tf_dialect_(context->getLoadedDialect<TensorFlowDialect>()), in ShapeInference()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emission_utils.cc | 658 CHECK(op->getDialect() == op->getContext()->getLoadedDialect("lmhlo")); in PartitionLmhloOperandsAndOutputs()
676 if (op->getDialect() == op->getContext()->getLoadedDialect("lmhlo")) { in GetHloOperands()
685 if (op->getDialect() == op->getContext()->getLoadedDialect("mhlo")) { in GetHloOperands()
696 if (op->getDialect() == op->getContext()->getLoadedDialect("lmhlo")) { in GetHloOutputs()
704 if (op->getDialect() == op->getContext()->getLoadedDialect("mhlo")) { in GetHloOutputs()
|
| D | ir_emitter_unnested.cc | 235 CHECK(op->getDialect() == op->getContext()->getLoadedDialect("lmhlo")); in LmhloOpIsElementwise()
258 CHECK(op->getDialect() == op->getContext()->getLoadedDialect("lmhlo")); in MayPreventVectorization()
267 CHECK(instr.getDialect() == instr.getContext()->getLoadedDialect("mhlo")) in MayPreventVectorization()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | constant_utils.cc | 49 auto dialect = etype.getContext()->getLoadedDialect("tf"); in CreateConstOpWithSingleValue()
|
| D | perception_ops_utils.cc | 40 return OpaqueElementsAttr::get(builder->getContext()->getLoadedDialect("tfl"), in CustomOption()
|
| D | nms_utils.cc | 35 return OpaqueElementsAttr::get(builder->getContext()->getLoadedDialect("tfl"), in CustomOption()
|
| D | tftext_utils.cc | 60 return OpaqueElementsAttr::get(builder->getContext()->getLoadedDialect("tfl"), in CustomOption()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/passes/ |
| D | canonicalize.cc | 157 mlir::Dialect *tf = context->getLoadedDialect<mlir::TF::TensorFlowDialect>(); in populateCanonicalizationPatterns()
|
| /external/llvm-project/mlir/test/lib/Dialect/SPIRV/ |
| D | TestAvailability.cpp | 33 Dialect *spvDialect = getContext().getLoadedDialect("spv"); in runOnFunction()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_operator.cc | 252 OpaqueElementsAttr::get(builder.getContext()->getLoadedDialect("tfl"), in CustomOptionsToAttributes()
|
| D | flatbuffer_import.cc | 554 auto dialect = elem_type.getContext()->getLoadedDialect("tf"); in BuildConstOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | convert_tensor.cc | 164 auto* dialect = builder->getContext()->getLoadedDialect("tf"); in ConvertTensor()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | export_graphdef.cc | 712 auto tf_dialect = module.getContext()->getLoadedDialect("tf"); in Convert()
785 Dialect* tf_dialect = func.getContext()->getLoadedDialect("tf"); in ConvertMlirFunctionToFunctionLibraryDef()
|
| /external/llvm-project/mlir/lib/Parser/ |
| D | Parser.cpp | 775 if (!getContext()->getLoadedDialect(dialectName) && in parseGenericOperation()
1614 if (!getContext()->getLoadedDialect(dialectName) && in parseCustomOperation()
|
| /external/llvm-project/mlir/lib/Target/LLVMIR/ |
| D | ModuleTranslation.cpp | 308 ompDialect(module->getContext()->getLoadedDialect("omp")), in ModuleTranslation()
|