| /external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/ |
| D | reorder_assert.cc | 41 for (auto func_op : module.getOps<mlir::func::FuncOp>()) { in runOnOperation() local
46 void ProcessFunction(mlir::func::FuncOp func_op) { in ProcessFunction()
83 bool IsFunctionNonSideEffectingOrAssert(mlir::func::FuncOp func_op) { in IsFunctionNonSideEffectingOrAssert()
|
| D | corert_converter.cc | 170 func::FuncOp func_op = op->getParentOfType<mlir::func::FuncOp>(); in ConvertOpHandler() local
181 mlir::func::FuncOp func_op = op->getParentOfType<mlir::func::FuncOp>(); in GetDistributedContext() local
198 mlir::func::FuncOp func_op = op->getParentOfType<mlir::func::FuncOp>(); in GetRemoteChainManager() local
219 auto func_op = op->getParentOfType<mlir::func::FuncOp>(); in GetLocalSideEffectChain() local
256 mlir::func::FuncOp func_op = op->getParentOfType<mlir::func::FuncOp>(); in GetTaskHandle() local
|
| D | set_shape_invariant_in_while_ops.cc | 32 mlir::func::FuncOp func_op = getOperation(); in MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID() local
|
| D | optimize_tf_control_flow_side_effect.cc | 26 mlir::func::FuncOp func_op, in FunctionHasSideEffect()
|
| D | insert_tensor_copy.cc | 48 mlir::func::FuncOp func_op = getOperation(); in MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/analysis/ |
| D | tensor_array_side_effect_analysis.cc | 32 mlir::func::FuncOp func_op) { in FunctionContainsOnlyNoSideEffectOpOrTensorArrayOp()
44 for (auto func_op : module.getOps<mlir::func::FuncOp>()) { in TensorArraySideEffectAnalysis() local
|
| D | tensor_array_side_effect_analysis.h | 39 bool HasAtMostTensorArrayEffect(mlir::func::FuncOp func_op) const { in HasAtMostTensorArrayEffect()
|
| D | test_tensor_array_side_effect_analysis.cc | 41 for (auto func_op : module.getOps<mlir::func::FuncOp>()) { in runOnOperation() local
|
| D | test_cost_analysis_pass.cc | 34 auto func_op = getOperation(); in runOnOperation() local
|
| D | cost_analysis.h | 39 explicit CostAnalysis(mlir::func::FuncOp func_op) { in CostAnalysis()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | convert_to_legacy_compile_and_replicate_attributes.cc | 43 LogicalResult ConvertToLegacyAttributes(func::FuncOp func_op) { in ConvertToLegacyAttributes()
58 func::FuncOp func_op = getOperation(); in runOnOperation() local
|
| D | resource_device_inference.cc | 72 func::FuncOp func_op, in PerFunctionResult()
151 LogicalResult ComputeResourceDevicesInComputation(func::FuncOp func_op, in ComputeResourceDevicesInComputation()
236 for (auto func_op : module.getOps<func::FuncOp>()) { in runOnOperation() local
271 auto func_op = worklist.pop_back_val(); in runOnOperation() local
|
| D | tpu_update_embedding_enqueue_op_inputs.cc | 66 func::FuncOp func_op, llvm::StringMap<Operation*>* enqueue_op_map, in FindTPUEmbeddingOps()
145 auto func_op = getOperation(); in runOnOperation() local
|
| D | cluster_tf_ops_pass.cc | 112 func::FuncOp func_op) { in GetFunctionMetadatas()
209 func::FuncOp func_op = func::FuncOp::create(loc, func_name, func_type); in CreateFunctions() local
307 for (auto func_op : module_op.getOps<func::FuncOp>()) { in runOnOperation() local
310 for (auto func_op : original_func) { in runOnOperation() local
|
| D | strip_noinline_attribute.cc | 31 for (auto func_op : getOperation().getOps<func::FuncOp>()) in runOnOperation() local
|
| D | canonicalize_compile_and_replicate_attributes.cc | 45 func::FuncOp func_op = getOperation(); in runOnOperation() local
|
| D | mark_initialized_variables_pass.cc | 59 auto func_op = getFunction(); in runOnFunction() local
|
| D | executor_tpuv1_inline_tpu_island.cc | 82 for (func::FuncOp func_op : in runOnOperation() local
|
| D | tensor_device_copy_conversion.cc | 49 func::FuncOp func_op = getOperation(); in runOnOperation() local
|
| D | initialize_variables_in_session_init.cc | 117 for (auto func_op : module.getOps<func::FuncOp>()) { in InitializeVariablesInSessionInitializer() local
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_function_test.cc | 410 TF_Operation* func_op = Use({}); in TEST_F() local
431 TF_Operation* func_op = Use({func_feed}); in TEST_F() local
452 TF_Operation* func_op = Use({func_feed}); in TEST_F() local
473 TF_Operation* func_op = Use({func_feed}); in TEST_F() local
492 TF_Operation* func_op = Use({func_feed}); in TEST_F() local
517 TF_Operation* func_op = Use({two, func_feed}); in TEST_F() local
542 TF_Operation* func_op = Use({two, func_feed}); in TEST_F() local
565 TF_Operation* func_op = Use({two, func_feed}); in TEST_F() local
617 TF_Operation* func_op = Use({two, ten, func_feed}); in TEST_F() local
648 TF_Operation* func_op = Use({two, func_feed}); in TEST_F() local
[all …]
|
| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | function_renaming.cc | 50 for (mlir::func::FuncOp func_op : in runOnOperation() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/transforms/ |
| D | tf_jitrt_peel_tiled_loops.cc | 62 auto func_op = getOperation(); in runOnOperation() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/analysis/ |
| D | side_effect_analysis.h | 60 SideEffectAnalysisInfo(func::FuncOp func_op, in SideEffectAnalysisInfo()
|
| /external/tensorflow/tensorflow/core/ir/ |
| D | tf_op_registry_test.cc | 58 auto func_op = cast<GraphFuncOp>(&module->front()); in TEST() local
|