Home
last modified time | relevance | path

Searched defs:defining_op (Results 1 – 18 of 18) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dexecutor_tpuv1_island_coarsening.cc97 Operation* defining_op = operand.getDefiningOp(); in HasDataDependencyWithUnscheduledOp() local
118 Operation* defining_op = input.getDefiningOp(); in HasControlDependencyWithUnscheduledOp() local
H A Dtpu_extract_outside_compilation.cc318 if (auto* defining_op = v.getDefiningOp()) { in GetStaticExternalOperands() local
345 Operation* defining_op = v.getDefiningOp(); in GetAllExternalOperands() local
H A Dcluster_formation.cc77 Operation* defining_op = operand.getDefiningOp(); in CanMergeIntoCluster() local
H A Dcluster_tf_ops_pass.cc145 Operation *defining_op = value.getDefiningOp(); in GetFunctionMetadatas() local
/aosp_15_r20/external/tensorflow/tensorflow/core/transforms/toposort/
H A Dtoposort_pass.cc55 Operation *defining_op = operand.getDefiningOp(); in SortTopologically() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/
H A Dinsert_custom_aggregation_ops.cc85 Operation *defining_op = input.getDefiningOp(); in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/
H A Dfuse_tpu_compile_and_execute_ops.cc69 if (auto defining_op = in runOnOperation() local
H A Dcross_device_transfer.cc147 Operation *defining_op = arg.getDefiningOp(); in runOnOperation() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
H A Dconvert_tfl_uint8.cc261 auto defining_op = terminator->getOperand(i).getDefiningOp(); in convert_graph_uint8_tensor() local
/aosp_15_r20/external/tensorflow/tensorflow/core/ir/utils/
H A Dshape_inference_utils_test.cc168 Operation *defining_op = operand.getDefiningOp(); in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
H A Dbreakup-islands.cc245 auto defining_op = operand.getDefiningOp(); in FindSourcesAndSinksInIsland() local
H A Dexport_graphdef.cc413 Operation* defining_op = in GetControlRetNodes() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dlegalize_tf_with_tf2xla.cc689 auto defining_op = operand.getDefiningOp(); in GetExprForOperand() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir/transforms/runtime/
H A Drt_to_llvm.cc236 if (auto *defining_op = std::get<1>(tuple).getDefiningOp()) { in EncodeArguments() local
/aosp_15_r20/external/tensorflow/tensorflow/core/transforms/constant_folding/
H A Dpass.cc712 Operation *defining_op = operand.getDefiningOp(); in matchAndRewrite() local
2799 Operation *defining_op = operand.getDefiningOp(); in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/
H A Dflatbuffer_import.cc978 absl::flat_hash_map<int32_t, const tflite::OperatorT*> defining_op; in PruneSubgraph() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dlayout_propagation_v2.cc1262 mlir::Operation* defining_op = cur_val.getDefiningOp(); in FindRoot() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
H A Dtfl_ops.cc116 auto *defining_op = operand.getDefiningOp(); in getDefiningBroadcastArgsOp() local