/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
H A D | executor_tpuv1_island_coarsening.cc | 97 Operation* defining_op = operand.getDefiningOp(); in HasDataDependencyWithUnscheduledOp() local 118 Operation* defining_op = input.getDefiningOp(); in HasControlDependencyWithUnscheduledOp() local
|
H A D | tpu_extract_outside_compilation.cc | 318 if (auto* defining_op = v.getDefiningOp()) { in GetStaticExternalOperands() local 345 Operation* defining_op = v.getDefiningOp(); in GetAllExternalOperands() local
|
H A D | cluster_formation.cc | 77 Operation* defining_op = operand.getDefiningOp(); in CanMergeIntoCluster() local
|
H A D | cluster_tf_ops_pass.cc | 145 Operation *defining_op = value.getDefiningOp(); in GetFunctionMetadatas() local
|
/aosp_15_r20/external/tensorflow/tensorflow/core/transforms/toposort/ |
H A D | toposort_pass.cc | 55 Operation *defining_op = operand.getDefiningOp(); in SortTopologically() local
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/ |
H A D | insert_custom_aggregation_ops.cc | 85 Operation *defining_op = input.getDefiningOp(); in matchAndRewrite() local
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/ |
H A D | fuse_tpu_compile_and_execute_ops.cc | 69 if (auto defining_op = in runOnOperation() local
|
H A D | cross_device_transfer.cc | 147 Operation *defining_op = arg.getDefiningOp(); in runOnOperation() local
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
H A D | convert_tfl_uint8.cc | 261 auto defining_op = terminator->getOperand(i).getDefiningOp(); in convert_graph_uint8_tensor() local
|
/aosp_15_r20/external/tensorflow/tensorflow/core/ir/utils/ |
H A D | shape_inference_utils_test.cc | 168 Operation *defining_op = operand.getDefiningOp(); in TEST_F() local
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
H A D | breakup-islands.cc | 245 auto defining_op = operand.getDefiningOp(); in FindSourcesAndSinksInIsland() local
|
H A D | export_graphdef.cc | 413 Operation* defining_op = in GetControlRetNodes() local
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
H A D | legalize_tf_with_tf2xla.cc | 689 auto defining_op = operand.getDefiningOp(); in GetExprForOperand() local
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir/transforms/runtime/ |
H A D | rt_to_llvm.cc | 236 if (auto *defining_op = std::get<1>(tuple).getDefiningOp()) { in EncodeArguments() local
|
/aosp_15_r20/external/tensorflow/tensorflow/core/transforms/constant_folding/ |
H A D | pass.cc | 712 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 D | flatbuffer_import.cc | 978 absl::flat_hash_map<int32_t, const tflite::OperatorT*> defining_op; in PruneSubgraph() local
|
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/ |
H A D | layout_propagation_v2.cc | 1262 mlir::Operation* defining_op = cur_val.getDefiningOp(); in FindRoot() local
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
H A D | tfl_ops.cc | 116 auto *defining_op = operand.getDefiningOp(); in getDefiningBroadcastArgsOp() local
|