Home
last modified time | relevance | path

Searched refs:argument_index (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtf_saved_model_freeze_variables.cc100 Operation* user_op, int argument_index, ElementsAttr value, in PropagateUsage() argument
108 (*arguments_to_erase)[call].push_back(argument_index); in PropagateUsage()
110 (*arguments_to_erase)[func].push_back(argument_index); in PropagateUsage()
111 work_list->push_back(std::make_pair(&func.getRegion(), argument_index)); in PropagateUsage()
114 (*arguments_to_erase)[if_op].push_back(argument_index); in PropagateUsage()
116 (*arguments_to_erase)[callee].push_back(argument_index - 1); in PropagateUsage()
118 std::make_pair(&callee.getBody(), argument_index - 1)); in PropagateUsage()
121 (*arguments_to_erase)[if_op].push_back(argument_index); in PropagateUsage()
123 work_list->push_back(std::make_pair(callee, argument_index)); in PropagateUsage()
126 (*arguments_to_erase)[while_op].push_back(argument_index); in PropagateUsage()
[all …]
/external/tensorflow/tensorflow/dtensor/mlir/
Dspmd_expansion.cc62 void UpdateFunctionInputShape(const int argument_index, in UpdateFunctionInputShape() argument
67 input_types[argument_index] = new_arg_type; in UpdateFunctionInputShape()
72 .getArgument(argument_index) in UpdateFunctionInputShape()
73 .setType(function.getFunctionType().getInput(argument_index)); in UpdateFunctionInputShape()
193 for (int argument_index = 0; argument_index < function.getNumArguments(); in UpdateFunctionArgsUsingLayout() local
194 ++argument_index) { in UpdateFunctionArgsUsingLayout()
196 argument_index, kCustomDeviceAttr); in UpdateFunctionArgsUsingLayout()
206 function.getFunctionType().getInput(argument_index)); in UpdateFunctionArgsUsingLayout()
211 if (mlir::failed(UpdateResourceArgumentType(argument_index, function))) in UpdateFunctionArgsUsingLayout()
218 .getInput(argument_index) in UpdateFunctionArgsUsingLayout()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/
Dfold_constants_to_subgraph.cc88 void CopyConstantIntoFunc(int argument_index, Operation* const_op, in CopyConstantIntoFunc() argument
97 func.getArgument(argument_index) in CopyConstantIntoFunc()
145 int argument_index = -1; in runOnOperation() local
148 argument_index = i; in runOnOperation()
156 CopyConstantIntoFunc(argument_index, op, func); in runOnOperation()
/external/deqp-deps/SPIRV-Tools/source/val/
Dvalidate_function.cpp259 for (size_t argument_index = 3, param_index = 2; in ValidateFunctionCall() local
260 argument_index < inst->operands().size(); in ValidateFunctionCall()
261 argument_index++, param_index++) { in ValidateFunctionCall()
262 const auto argument_id = inst->GetOperandAs<uint32_t>(argument_index); in ValidateFunctionCall()
266 << "Missing argument " << argument_index - 3 << " definition."; in ValidateFunctionCall()
272 << "Missing argument " << argument_index - 3 in ValidateFunctionCall()
/external/swiftshader/third_party/SPIRV-Tools/source/val/
Dvalidate_function.cpp259 for (size_t argument_index = 3, param_index = 2; in ValidateFunctionCall() local
260 argument_index < inst->operands().size(); in ValidateFunctionCall()
261 argument_index++, param_index++) { in ValidateFunctionCall()
262 const auto argument_id = inst->GetOperandAs<uint32_t>(argument_index); in ValidateFunctionCall()
266 << "Missing argument " << argument_index - 3 << " definition."; in ValidateFunctionCall()
272 << "Missing argument " << argument_index - 3 in ValidateFunctionCall()
/external/angle/third_party/vulkan-deps/spirv-tools/src/source/val/
Dvalidate_function.cpp260 for (size_t argument_index = 3, param_index = 2; in ValidateFunctionCall() local
261 argument_index < inst->operands().size(); in ValidateFunctionCall()
262 argument_index++, param_index++) { in ValidateFunctionCall()
263 const auto argument_id = inst->GetOperandAs<uint32_t>(argument_index); in ValidateFunctionCall()
267 << "Missing argument " << argument_index - 3 << " definition."; in ValidateFunctionCall()
273 << "Missing argument " << argument_index - 3 in ValidateFunctionCall()
/external/tensorflow/tensorflow/cc/experimental/libtf/
Dobject.h637 int argument_index, in Call() argument
653 int argument_index, in Call() argument
656 Handle h(std::move(args_in.tuple()[argument_index])); in Call()
661 std::to_string(argument_index) + in Call()
664 name, fn, argument_index + 1, args_in, args..., *x); in Call()
/external/bcc/src/cc/
Dbcc_usdt.h72 int location_index, int argument_index,
/external/bcc/src/cc/includes/
Dbcc_usdt.h72 int location_index, int argument_index,
/external/bcc/src/cc/usdt/
Dusdt.cc593 int location_index, int argument_index, in bcc_usdt_get_argument() argument
599 if (argument_index < 0 || (size_t)argument_index >= probe->num_arguments()) in bcc_usdt_get_argument()
604 auto const &arg = location.arguments_[argument_index]; in bcc_usdt_get_argument()
/external/minijail/tools/
Dparser.py358 argument_index = int(argument.value[3:])
359 if not (0 <= argument_index < bpf.MAX_SYSCALL_ARGUMENTS):
361 return Atom(argument_index, operator.value, value)
Dcompiler.py342 block = bpf.Atom(atom.argument_index, atom.op, atom.value,
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc646 int argument_index; member
649 return std::tie(replica_id, argument_index) < in operator <()
650 std::tie(rhs.replica_id, rhs.argument_index); in operator <()
656 int argument_index; member
658 return std::tie(host_device, argument_index) < in operator <()
659 std::tie(rhs.host_device, rhs.argument_index); in operator <()
662 return (argument_index == rhs.argument_index) && in operator ==()
/external/deqp/external/openglcts/modules/gl/
Dgl4cGPUShaderFP64Tests.cpp14798 glw::GLuint argument_index[3] = { 0 }; in prepareComponents() local
14866 argument_index[i] = (vertex / argument_step[i]) % argument_reset[i]; in prepareComponents()
14875 double_argument_start[argument] += argument_index[argument]; in prepareComponents()
14901 uint_argument_start += argument_index[argument]; in prepareComponents()
14915 uint_argument_start += argument_index[argument]; in prepareComponents()