| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | materialize_mlir_passthrough_op.cc | 70 if (main.getFunctionType().getNumResults() != op->getNumResults()) { in runOnOperation()
72 << op->getNumResults() in runOnOperation()
74 << main.getFunctionType().getNumResults() in runOnOperation()
|
| D | convert_control_to_data_outputs.cc | 153 if (fetch.getNumOperands() == graph_op->getNumResults()) return false; in RemoveAllControlOutputs()
157 fetch.fetches().drop_front(graph_op->getNumResults())) { in RemoveAllControlOutputs()
164 graph_op.getNumResults(), in RemoveAllControlOutputs()
165 fetch.getNumOperands() - graph_op.getNumResults()); in RemoveAllControlOutputs()
419 int num_old_outputs = while_body.getNumResults(); in ConvertControlToDataOutputs()
|
| D | functional_control_flow_to_cfg.cc | 120 assert(op->getNumResults() == block->getNumArguments()); in ReplaceOpResultWithBlockArgs()
121 for (unsigned i = 0, e = op->getNumResults(); i != e; ++i) { in ReplaceOpResultWithBlockArgs()
251 assert(cond_call_op->getNumResults() == 1); in LowerWhileOp()
|
| D | tpu_merge_variables_with_execute.cc | 257 llvm::SmallVector<bool, 8> output_merged(execute_launch.getNumResults(), in BuildVariableAccessInfo()
330 execute_launch.getNumResults()); in BuildVariableAccessInfo()
331 for (int i = 0, end = execute_launch.getNumResults(); i < end; ++i) { in BuildVariableAccessInfo()
370 const int parallel_execute_num_results = parallel_execute_op->getNumResults(); in ReplaceParallelExecute()
399 new_parallel_execute_op->getNumResults(); in ReplaceParallelExecute()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_remaining_ops.cc | 113 if (op->getNumResults() != func.getFunctionType().getNumResults()) in verify()
115 << func.getFunctionType().getNumResults() in verify()
117 << op->getNumResults() in verify()
|
| D | tf_executor.cc | 188 if (fetch.getNumOperands() < graph.getNumResults()) in verify()
194 const int64_t num_results = graph.getNumResults(); in verify()
272 return wrapped_op.getNumResults() == yield.getNumOperands() && in WrapsSingleOp()
288 int result_count = island.getNumResults() - 1; // -1 for the control token in verify()
303 for (int operand_idx : llvm::seq<int>(0, island.getNumResults() - 1)) { in verify()
461 if (switchn.getNumResults() != num_outs.getInt() + 1) in verify()
464 << (switchn.getNumResults() - 1); in verify()
512 p << " of " << (getNumResults() - 1); in print()
1006 if (op.getNumOperands() != 0 || op.getNumResults() != 1 || in matchAndRewrite()
1029 if (op.getNumOperands() != 0 || op.getNumResults() != 2 || in matchAndRewrite()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | lstm_utils_test.cc | 132 EXPECT_EQ(fused_lstm_func_.getFunctionType().getNumResults(), 1); in TEST_F()
161 EXPECT_EQ(it->getNumResults(), 1); in TEST_F()
183 EXPECT_EQ(fused_lstm_func_.getFunctionType().getNumResults(), 1); in TEST_F()
215 EXPECT_EQ(it->getNumResults(), 1); in TEST_F()
232 EXPECT_EQ(fused_ln_lstm_func_.getFunctionType().getNumResults(), 1); in TEST_F()
242 EXPECT_EQ(it->getNumResults(), 1); in TEST_F()
254 EXPECT_EQ(fused_ln_lstm_func_.getFunctionType().getNumResults(), 1); in TEST_F()
|
| D | perception_ops_utils.cc | 118 if (func_.getFunctionType().getNumResults() != 1) { in VerifySignature()
121 << func_.getFunctionType().getNumResults(); in VerifySignature()
214 if (func_.getFunctionType().getNumResults() != 1) { in VerifySignature()
217 << func_.getFunctionType().getNumResults(); in VerifySignature()
|
| D | nms_utils.cc | 69 if (func_.getFunctionType().getNumResults() != 2) { in VerifySignature()
72 << func_.getFunctionType().getNumResults(); in VerifySignature()
188 if (func_.getFunctionType().getNumResults() != 4) { in VerifySignature()
191 << ": " << func_.getFunctionType().getNumResults(); in VerifySignature()
|
| D | tftext_utils.cc | 101 if (func.getNumResults() != kValidNumOfOutput[input_type.getRank()]) { in VerifyWhitespaceTokenizer()
112 if (func.getNumResults() > 1) { in VerifyWhitespaceTokenizer()
119 if (func.getNumResults() > 2) { in VerifyWhitespaceTokenizer()
275 func.getFunctionType().getNumResults() != 1) { in VerifySgnnProjection()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/quantization/ |
| D | quantization_context.cc | 56 for (int res = 0, e = op.getNumResults(); res != e; ++res) { in QuantizeContext()
76 for (int i = 0; i < op.getNumResults(); ++i) { in GetSignature()
145 for (int res = 0, e = op.getNumResults(); res != e; ++res) { in Finalize()
176 for (auto i = 0; i < op.getNumResults(); ++i) { in DumpStates()
217 for (int i = 0, e = op->getNumResults(); i != e; ++i) { in GetQuantParamsForSameScaleConstraint()
229 if (op->getNumResults() == 1 && immutable_results_num == 1) { in GetQuantParamsForSameScaleConstraint()
248 if (op->getNumResults() == 1 && mutable_results_num == 1) { in GetQuantParamsForSameScaleConstraint()
272 for (int res = 0, e = op->getNumResults(); res != e; ++res) { in PropagateQuantParams()
|
| /external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/ |
| D | insert_main_function.cc | 97 for (int i = 0; i < func.getNumResults(); ++i) { in SetFunctionPrivate()
169 output_names.size() != main_func.getNumResults()) { in CreateMainFunction()
175 << ", got: " << main_func.getNumResults(); in CreateMainFunction()
187 int numResults = main_func.getNumResults(); in CreateMainFunction()
210 result_types.begin() + result_idx, function.getNumResults()); in CreateMainFunction()
211 result_idx += function.getNumResults(); in CreateMainFunction()
|
| /external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| D | split_spmd_expander.cc | 103 llvm::DenseMap<int, Layout> output_layouts(split_op.getNumResults()); in ComputeLayoutForward()
106 for (int i = 0; i < split_op.getNumResults(); ++i) { in ComputeLayoutForward()
156 llvm::DenseMap<int, Layout> output_layouts(split_v_op.getNumResults()); in ComputeLayoutForward()
159 for (int i = 0; i < split_v_op.getNumResults(); ++i) { in ComputeLayoutForward()
|
| D | trivial_spmd_expander.cc | 39 assert(output_types.size() == parent_op->getNumResults()); in ExpandOp()
84 llvm::DenseMap<int, Layout> output_layouts(op->getNumResults()); in ComputeLayoutForward()
|
| D | save_restore_spmd_expander.cc | 574 llvm::SmallVector<std::string> new_shapes_and_slices(op->getNumResults()); in ExpandRestoreV2OpHelper()
721 global_shapes.reserve(op->getNumResults()); in ExpandRestoreV2Op()
737 if (op->getNumResults() == 1 && !GetShapeOfValue(op->getResult(0)).ok()) { in ExpandRestoreV2Op()
752 new_types.reserve(op->getNumResults()); in ExpandRestoreV2Op()
794 llvm::SmallVector<Layout> layouts(restore_op->getNumResults()); in GetLayoutsFromAssignVariableOps()
863 if (layouts.size() != restore_v2.getNumResults()) { in ComputeLayoutForward()
866 restore_v2.getNumResults(), in ComputeLayoutForward()
870 llvm::DenseMap<int, Layout> output_layouts(restore_v2.getNumResults()); in ComputeLayoutForward()
884 llvm::DenseMap<int, Layout> output_layouts(restore_v2.getNumResults()); in ComputeLayoutForward()
|
| D | replicated_spmd_expander.cc | 121 llvm::DenseMap<int, Layout> output_layouts(op->getNumResults()); in ComputeLayoutForward()
122 for (int i = 0; i < op->getNumResults(); ++i) { in ComputeLayoutForward()
|
| D | nullary_spmd_expander.cc | 41 if (op->getNumResults() == 0) return op; in ExpandOp()
121 for (auto i = 0; i < op->getNumResults(); ++i) { in ComputeLayoutForward()
|
| /external/tensorflow/tensorflow/core/transforms/func_to_graph/ |
| D | func_to_graph.cc | 83 if (result_index >= op->getNumResults()) { in FuncToGraph()
87 op->getNumResults(), " results"); in FuncToGraph()
|
| /external/tensorflow/tensorflow/compiler/xla/mlir/transforms/runtime/ |
| D | calling_convention.cc | 57 results.reserve(func.getNumResults()); in DefaultCallingConvention()
82 inputs.reserve(1 + func.getNumInputs() + func.getNumResults()); in ResultsToOutsCallingConvention()
|
| /external/tensorflow/tensorflow/core/ir/ |
| D | tf_op_wrapper.h | 87 Value controlRet() { return op_->getResult(op_->getNumResults() - 1); } in controlRet()
91 return op_->getResults().slice(0, op_->getNumResults() - 1); in getNonControlResults()
|
| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | shape_utils.cc | 93 const int num_results = op->getNumResults(); in ExtractGlobalOutputShape()
96 if (output_index >= op->getNumResults()) in ExtractGlobalOutputShape()
213 assert(inferred_return_types.size() == op->getNumResults()); in InferSPMDExpandedLocalShape()
|
| D | spmd_expander.cc | 73 if (computed_layout.empty() && op->getNumResults() != 0) { in ExpandOpAndSetLayout()
83 global_output_shapes.reserve(op->getNumResults()); in ExpandOpAndSetLayout()
|
| /external/tensorflow/tensorflow/core/transforms/graph_compactor/ |
| D | pass.cc | 116 if (func.getNumResults()) { in runOnOperation()
119 res_attrs.reserve(func.getNumResults()); in runOnOperation()
127 if (func.getNumArguments() || func.getNumResults()) { in runOnOperation()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | outline_with_xla_framework.cc | 82 op->getNumResults() != 0) in matchAndRewrite()
91 SmallVector<Type> result_array(func.getFunctionType().getNumResults(), in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/gml_st/transforms/ |
| D | fusion.cc | 65 assert(materializeOp->getNumResults() == 1 && "assume single result"); in matchAndRewrite()
74 if (genericOp.getNumResults() != 1 || !genericOp.hasTensorSemantics()) { in matchAndRewrite()
|