Home
last modified time | relevance | path

Searched refs:applyTensorflowAndCLOptions (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/
Dkernel_creator.cc200 applyTensorflowAndCLOptions(pm); in LowerTFtoLoops()
262 applyTensorflowAndCLOptions(pm); in LowerLoopsToGPUorCPU()
360 tensorflow::applyTensorflowAndCLOptions(pm); in LowerKernelBodiesToLowLevelIr()
380 applyTensorflowAndCLOptions(pm); in AmendKernelLLVMIRWithStaticKnowledge()
398 applyTensorflowAndCLOptions(pm); in GenerateDeviceCode()
415 applyTensorflowAndCLOptions(pm); in LowerHostSideToFinalForm()
/external/tensorflow/tensorflow/compiler/xla/service/mlir_gpu/
Dkernel_lowering.cc52 tensorflow::applyTensorflowAndCLOptions(pm); in LowerLHLOToGPU()
182 tensorflow::applyTensorflowAndCLOptions(pm); in LowerKernelBodiesToNVVM()
243 tensorflow::applyTensorflowAndCLOptions(pm); in LowerKernelBodiesToROCDL()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Ddump_mlir_util.h81 void applyTensorflowAndCLOptions(mlir::PassManager& pm,
Ddump_mlir_util.cc260 void applyTensorflowAndCLOptions(mlir::PassManager& pm, in applyTensorflowAndCLOptions() function
Dcompile_mlir_util.cc345 applyTensorflowAndCLOptions(tf2xla); in LegalizeToHlo()
602 applyTensorflowAndCLOptions(pm); in CompileGraphSetup()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dgraph_optimization_pass.cc45 ::tensorflow::applyTensorflowAndCLOptions(pm); in Run()
Dbridge.cc61 ::tensorflow::applyTensorflowAndCLOptions(bridge); in RunTPUBridge()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/c/
Dc_api_unified_experimental_mlir.cc525 ::tensorflow::applyTensorflowAndCLOptions(pm); in GetFunctionDef()