Home
last modified time | relevance | path

Searched refs:getNumArguments (Results 1 – 25 of 85) sorted by relevance

1234

/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/
Dinsert_main_function.cc89 for (int i = 0; i < func.getNumArguments(); ++i) { in SetFunctionPrivate()
168 if (input_names.size() != main_func.getNumArguments() || in CreateMainFunction()
173 << input_names.size() << ", got: " << main_func.getNumArguments() in CreateMainFunction()
179 int numArgs = main_func.getNumArguments(); in CreateMainFunction()
207 main_func.getArguments().begin() + arg_idx, function.getNumArguments()); in CreateMainFunction()
208 arg_idx += function.getNumArguments(); in CreateMainFunction()
Dprepare_quantize.cc123 return func.getNumArguments() == quant_specs_.input_ranges.size(); in IsLegalQuantSpecs()
222 for (int i = 0, e = func.getNumArguments(); i != e; ++i) { in SetInputNodesQuantizationParams()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Doptimize_global_tensors.cc88 for (size_t i = 0, e = func.getNumArguments(); i < e; i++) { in CreateGlobalTensorUsesMap()
139 llvm::BitVector args_to_erase(func.getNumArguments()); in EraseUnusedBoundInputs()
140 for (int i = 0, e = func.getNumArguments(); i < e; i++) { in EraseUnusedBoundInputs()
Ddeduplicate_bound_input_bindings.cc40 llvm::BitVector arg_indices_to_erase(func.getNumArguments()); in runOnOperation()
41 for (unsigned i = 0, e = func.getNumArguments(); i < e; i++) { in runOnOperation()
Dmaterialize_mlir_passthrough_op.cc63 if (main.getNumArguments() != op->getNumOperands()) { in runOnOperation()
67 << main.getNumArguments() << " args)\n"; in runOnOperation()
Dfreeze_saved_model_assets.cc61 llvm::BitVector args_to_erase(func.getNumArguments()); in runOnOperation()
64 for (int i = 0, e = func.getNumArguments(); i < e; ++i) { in runOnOperation()
Dpromote_resources_to_args.cc181 llvm::SmallVector<ResourceInfo, 4> resources(function.getNumArguments()); in PromoteResourcesToArguments()
198 function.getNumArguments() - var_handle_shared_names.size()); in PromoteResourcesToArguments()
272 function.getNumArguments() - var_handle_shared_names.size(); in PromoteResourcesToArguments()
400 function.getNumArguments() - var_handle_shared_names.size(); in runOnOperation()
Dtf_saved_model_freeze_variables.cc374 llvm::BitVector args_to_erase_bit_vector(func.getNumArguments()); in FreezeVariables()
398 new_while_op.body().front().getNumArguments()); in FreezeVariables()
402 new_while_op.cond().front().getNumArguments()); in FreezeVariables()
/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dperception_ops_utils.cc113 if (func_.getNumArguments() != 2) { in VerifySignature()
116 << func_.getNumArguments(); in VerifySignature()
209 if (func_.getNumArguments() != 2) { in VerifySignature()
212 << func_.getNumArguments(); in VerifySignature()
Dnms_utils.cc63 if (func_.getNumArguments() < 5) { in VerifySignature()
67 << func_.getNumArguments(); in VerifySignature()
183 if (func_.getNumArguments() != 3) { in VerifySignature()
186 << ": " << func_.getNumArguments(); in VerifySignature()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dxla_call_module_op.cc142 nr_dim_args + ctx->num_inputs() == main_body.getNumArguments(), in RefineDynamicShapes()
145 "The module expects ", main_body.getNumArguments(), in RefineDynamicShapes()
152 std::vector<mlir::Type> static_input_types(main_body.getNumArguments()); in RefineDynamicShapes()
183 for (auto i = 0; i < main_body.getNumArguments(); ++i) { in RefineDynamicShapes()
/external/tensorflow/tensorflow/compiler/xla/mlir/utils/runtime/
Dconstraints.cc37 constraints.reserve(func.getNumArguments()); in GetArgumentsConstraints()
51 for (int i = 0; i < func.getNumArguments(); ++i) { in GetArgumentsConstraints()
/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/
Dremove_tf_if_const_args.cc111 llvm::BitVector const_arg_indices_bv(branch.getNumArguments()); in CreateBranchFunction()
131 llvm::SmallVector<mlir::Value, 4> call_args(branch.getNumArguments()); in CreateBranchFunction()
146 assert(j < block->getNumArguments()); in CreateBranchFunction()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf_control_flow.cc87 llvm::to_vector(llvm::seq<unsigned>(0, region.getNumArguments()))); in ReplaceBlockArgumentsWithImplicitOperands()
114 detupled_args.reserve(func.getNumArguments()); in ImportXlaRegion()
116 for (int64_t i = 0, s = func.getNumArguments(); i < s; i++) { in ImportXlaRegion()
211 for (int i = block->getNumArguments() - 2; i >= 0; --i) in ReplaceBlockArgs()
247 assert(block->getNumArguments() == 1); in ReplaceImplicitInputsWithTupleElements()
278 assert(block.getNumArguments() == 0); in TupleImplicitInputs()
/external/tensorflow/tensorflow/core/ir/
Dutility.cc40 return block.getArgument(data.getArgNumber() + block.getNumArguments() / 2); in GetLoopRegionControlOf()
44 return block.getArgument(ctl.getArgNumber() - block.getNumArguments() / 2); in GetLoopRegionDataOf()
/external/tensorflow/tensorflow/core/transforms/graph_compactor/
Dpass.cc104 if (func.getNumArguments()) { in runOnOperation()
107 arg_attrs.reserve(func.getNumArguments()); in runOnOperation()
109 for (int i = 0, e = func.getNumArguments(); i != e; i += 2) { in runOnOperation()
127 if (func.getNumArguments() || func.getNumResults()) { in runOnOperation()
/external/tensorflow/tensorflow/dtensor/mlir/
Dhandle_sparsetensors.cc220 {main_func.getArgument(main_func.getNumArguments() - 3), in runOnOperation()
221 main_func.getArgument(main_func.getNumArguments() - 2), in runOnOperation()
222 main_func.getArgument(main_func.getNumArguments() - 1), in runOnOperation()
Ddevice_utils.cc36 if (function.getNumArguments() == 0) in DeviceId()
/external/tensorflow/tensorflow/core/runtime_fallback/opdefs/
Dtfrt_fallback_util.cc39 for (int arg_index = 0; arg_index < func.getNumArguments(); ++arg_index) { in ForEachArgConsumedByFallback()
/external/tensorflow/tensorflow/compiler/mlir/tfrt/ir/
Dtfrt_fallback_util.cc39 for (int arg_index = 0; arg_index < func.getNumArguments(); ++arg_index) { in ForEachArgConsumedByFallback()
/external/llvm/tools/llvm-diff/
DDiffLog.cpp28 unsigned LogBuilder::getNumArguments() const { return Arguments.size(); } in getNumArguments() function in LogBuilder
DDiffLog.h56 unsigned getNumArguments() const;
/external/clang/lib/Lex/
DMacroArgs.cpp303 StringifiedArgs.resize(getNumArguments()); in getStringifiedArgument()
305 sizeof(StringifiedArgs[0])*getNumArguments()); in getStringifiedArgument()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_saved_model.cc229 for (int i = 0, e = func.getNumArguments(); i < e; i++) { in HasAnyTfSavedModelArgAttr()
333 for (int i = 0, e = func.getNumArguments(); i < e; i++) { in VerifyExportedFunc()
356 for (int i = 0, e = func.getNumArguments(); i < e; i++) { in VerifyExportedFunc()
/external/clang/include/clang/Lex/
DMacroArgs.h99 unsigned getNumArguments() const { return NumUnexpArgTokens; } in getNumArguments() function

1234