Home
last modified time | relevance | path

Searched refs:input_idx (Results 1 – 25 of 47) sorted by relevance

12

/external/autotest/site_utils/tester_feedback/
Dinput_handlers.py139 input_idx = int(input_str) - 1
140 if input_idx in range(len(self._choices)):
141 return input_idx, self._choices[input_idx]
/external/tensorflow/tensorflow/core/kernels/linalg/
Dlinalg_ops_common.cc178 for (int input_idx = 0; input_idx < context->num_inputs(); ++input_idx) { in PrepareOutputs() local
179 unused_inputs.insert(input_idx); in PrepareOutputs()
202 for (int input_idx : unused_inputs) { in PrepareOutputs() local
204 input_idx, output_idx, output_tensor_shape, &out)) { in PrepareOutputs()
206 unused_inputs.erase(input_idx); in PrepareOutputs()
/external/tensorflow/tensorflow/lite/tools/optimize/
Dquantize_model.cc106 const int32_t input_idx = input.first; in PopulateRealValueOpSet() local
107 const int32_t tensor_idx = op->inputs[input_idx]; in PopulateRealValueOpSet()
224 int32_t input_idx) { in InputQuantizeRequired() argument
228 if (std::find(op->inputs.begin(), op->inputs.end(), input_idx) != in InputQuantizeRequired()
239 const auto* tensor = subgraph->tensors[input_idx].get(); in InputQuantizeRequired()
405 const int32_t input_idx = SetInputType( in SetInputAndOutputTypes() local
407 if (input_idx < 0) { in SetInputAndOutputTypes()
410 subgraph->inputs[i] = input_idx; in SetInputAndOutputTypes()
472 for (size_t input_idx = 0; input_idx < op->inputs.size(); ++input_idx) { in ApplyConstraints() local
473 TensorT* input_tensor = subgraph->tensors[op->inputs[input_idx]].get(); in ApplyConstraints()
[all …]
Dmodify_model_interface.cc57 for (size_t input_idx = 0; input_idx < subgraph->inputs.size(); in GetInputTensors() local
58 input_idx++) { in GetInputTensors()
59 TensorT* tensor = subgraph->tensors[subgraph->inputs[input_idx]].get(); in GetInputTensors()
61 input_tensors.insert({tensor, input_idx}); in GetInputTensors()
400 for (size_t input_idx = 0; input_idx < subgraph->inputs.size(); in AddUint8Dequant() local
401 input_idx++) { in AddUint8Dequant()
402 const int32_t tensor_idx = subgraph->inputs[input_idx]; in AddUint8Dequant()
Dquantize_model_test.cc283 const int32_t input_idx = subgraph->inputs[0]; in TEST_P() local
286 EXPECT_EQ(quant_op->inputs[0], input_idx); in TEST_P()
292 EXPECT_EQ(subgraph->tensors[input_idx]->type, TensorType_FLOAT32); in TEST_P()
293 EXPECT_EQ(subgraph->tensors[input_idx]->name, "input"); in TEST_P()
306 if (input_idx != tensor_idx && output_idx != tensor_idx) { in TEST_P()
350 const int32_t input_idx = subgraph->inputs[0]; in TEST_P() local
353 EXPECT_EQ(quant_op_uint8_int8->inputs[0], input_idx); in TEST_P()
361 EXPECT_EQ(subgraph->tensors[input_idx]->type, TensorType_UINT8); in TEST_P()
362 EXPECT_EQ(subgraph->tensors[input_idx]->name, "input"); in TEST_P()
363 EXPECT_EQ(subgraph->tensors[input_idx]->quantization->scale.size(), 1); in TEST_P()
[all …]
/external/tensorflow/tensorflow/compiler/jit/
Dnode_matchers.cc172 for (int input_idx = 0, e = input_matchers->size(); input_idx < e; in MatchAndExplain() local
173 input_idx++) { in MatchAndExplain()
174 if (!MatchAndExplainInput(node, input_idx, listener)) { in MatchAndExplain()
303 bool MatchAndExplainInput(const Node* node, int input_idx, in MatchAndExplainInput()
306 if (!node->input_edge(input_idx, &edge).ok()) { in MatchAndExplainInput()
308 *listener << "\ncould not find incoming edge for input " << input_idx; in MatchAndExplainInput()
315 if ((*input_matchers)[input_idx].MatchAndExplain(input, &inner_listener)) { in MatchAndExplainInput()
320 *listener << "\ninput " << input_idx << " does not match expected:\n"; in MatchAndExplainInput()
321 (*input_matchers)[input_idx].DescribeTo(listener->stream()); in MatchAndExplainInput()
Dbuild_xla_ops_pass.cc387 int input_idx = 0; in GetXlaRunArgs() local
391 s.WithOpName("check_input_", input_idx), o, in GetXlaRunArgs()
392 absl::StrCat("CheckNumerics failed for input ", input_idx, "(", in GetXlaRunArgs()
398 input_idx++; in GetXlaRunArgs()
Dxla_launch_util.cc126 for (int input_idx = 0; input_idx < ctx->num_inputs(); input_idx++) { in InputsFromContext() local
127 inputs.push_back(&ctx->input(input_idx)); in InputsFromContext()
/external/tensorflow/tensorflow/core/common_runtime/
Dplacer_inspection_required_ops_utils.cc168 Status AddInputIdentity(Node* node, int input_idx, Graph* graph, in AddInputIdentity() argument
171 TF_RETURN_IF_ERROR(node->input_edge(input_idx, &edge)); in AddInputIdentity()
177 builder.Attr("T", node->input_type(input_idx)); in AddInputIdentity()
179 node->input_type(input_idx)); in AddInputIdentity()
187 << input_idx << " \n" in AddInputIdentity()
198 TF_RETURN_IF_ERROR(graph->UpdateEdge(identity_node, 0, node, input_idx)); in AddInputIdentity()
/external/tensorflow/tensorflow/core/kernels/
Daggregate_ops.cc57 for (int input_idx = 0; input_idx < num; ++input_idx) { in Compute() local
58 if (ctx->forward_input_to_output_with_shape(input_idx, 0, input0.shape(), in Compute()
60 reused_input = input_idx; in Compute()
Ddepthtospace_op_gpu.cu.cc69 GPU_1D_KERNEL_LOOP(input_idx, nthreads) { in D2S_NCHW()
77 const int n_bY_bX_oC_iY = input_idx / input_width; in D2S_NCHW()
78 const int iX = input_idx - n_bY_bX_oC_iY * input_width; in D2S_NCHW()
96 *(output_ptr + output_idx) = ldg(input_ptr + input_idx); in D2S_NCHW()
Dspacetodepth_op_gpu.cu.cc69 GPU_1D_KERNEL_LOOP(input_idx, nthreads) { in S2D_NCHW()
78 const int n_iC_oY_bY_oX = input_idx / block_size; in S2D_NCHW()
79 const int bX = input_idx - n_iC_oY_bY_oX * block_size; in S2D_NCHW()
95 *(output_ptr + output_idx) = ldg(input_ptr + input_idx); in S2D_NCHW()
Dstateful_random_ops.cc155 Status GetScalar(const Tensor& tensor, int input_idx, T* result) { in GetScalar() argument
158 return errors::InvalidArgument("input ", std::to_string(input_idx), in GetScalar()
163 return errors::InvalidArgument("dtype of input ", std::to_string(input_idx), in GetScalar()
172 Status GetAlg(OpKernelContext* ctx, int input_idx, Algorithm* alg) { in GetAlg() argument
174 TF_RETURN_IF_ERROR(GetScalar(ctx->input(input_idx), input_idx, &alg_id)); in GetAlg()
Dstateless_random_ops_v2.cc51 Status GetScalar(const Tensor& tensor, int input_idx, T* result) { in GetScalar() argument
54 return errors::InvalidArgument("input ", std::to_string(input_idx), in GetScalar()
59 return errors::InvalidArgument("dtype of input ", std::to_string(input_idx), in GetScalar()
/external/tensorflow/tensorflow/lite/objc/tests/
DTFLMetalDelegateTests.m90 for (int input_idx = 0; input_idx < kNumInputs; ++input_idx) {
91 TFLTensor* inputTensor = [interpreter inputTensorAtIndex:input_idx error:&error];
/external/tensorflow/tensorflow/core/framework/
Dshape_inference.cc718 int input_idx, ShapeHandle* out) { in MakeShapeFromShapeTensorTreatScalarAsUnknownShape() argument
720 TF_RETURN_IF_ERROR(WithRankAtMost(input(input_idx), 1, &input_shape)); in MakeShapeFromShapeTensorTreatScalarAsUnknownShape()
722 request_input_tensor_as_partial_shape(input_idx); in MakeShapeFromShapeTensorTreatScalarAsUnknownShape()
724 if (input_idx < input_tensors_as_shapes_size && in MakeShapeFromShapeTensorTreatScalarAsUnknownShape()
725 input_tensors_as_shapes_[input_idx].IsSet() && in MakeShapeFromShapeTensorTreatScalarAsUnknownShape()
726 RankKnown(input_tensors_as_shapes_[input_idx])) { in MakeShapeFromShapeTensorTreatScalarAsUnknownShape()
727 *out = input_tensors_as_shapes_[input_idx]; in MakeShapeFromShapeTensorTreatScalarAsUnknownShape()
733 input_tensor(input_idx), input_shape, out); in MakeShapeFromShapeTensorTreatScalarAsUnknownShape()
736 Status InferenceContext::MakeShapeFromShapeTensor(int input_idx, in MakeShapeFromShapeTensor() argument
739 TF_RETURN_IF_ERROR(WithRank(input(input_idx), 1, &input_shape)); in MakeShapeFromShapeTensor()
[all …]
/external/tensorflow/tensorflow/lite/tools/
Dverifier.cc540 for (const int input_idx : *op->inputs()) { in VerifySubGraphConsistency()
541 if (input_idx == kTfLiteOptionalTensor) continue; in VerifySubGraphConsistency()
542 if (constant_tensors.find(input_idx) == constant_tensors.end() && in VerifySubGraphConsistency()
543 variable_tensors.find(input_idx) == variable_tensors.end() && in VerifySubGraphConsistency()
544 subgraph_input_tensors.find(input_idx) == in VerifySubGraphConsistency()
546 output_tensors.find(input_idx) == output_tensors.end()) { in VerifySubGraphConsistency()
549 input_idx, op_idx, EnumNameBuiltinOperator(builtin_code)); in VerifySubGraphConsistency()
/external/tensorflow/tensorflow/core/grappler/
Dgraph_topology_view.cc125 const int input_idx = it->second; in InitializeFromGraph() local
126 fanins_[node_idx].push_back(input_idx); in InitializeFromGraph()
127 fanouts_[input_idx].push_back(node_idx); in InitializeFromGraph()
/external/tensorflow/tensorflow/lite/delegates/hexagon/
Dhexagon_delegate_kernel.cc97 for (int input_idx = 0; input_idx < node->inputs->size; ++input_idx) { in Eval() local
98 const auto tensor_index = node->inputs->data[input_idx]; in Eval()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dkernel_thunk.cc72 int input_idx = 0; in PrintBufferContents() local
83 VLOG(100) << "BUF(" << input_idx++ << ") = " << buffer_contents; in PrintBufferContents()
/external/tensorflow/tensorflow/lite/java/src/main/native/
Dnativeinterpreterwrapper_jni.cc629 jint input_idx, jintArray dims, jboolean strict) { in Java_org_tensorflow_lite_NativeInterpreterWrapper_resizeInput() argument
635 if (input_idx < 0 || input_idx >= interpreter->inputs().size()) { in Java_org_tensorflow_lite_NativeInterpreterWrapper_resizeInput()
639 input_idx, interpreter->inputs().size()); in Java_org_tensorflow_lite_NativeInterpreterWrapper_resizeInput()
642 const int tensor_idx = interpreter->inputs()[input_idx]; in Java_org_tensorflow_lite_NativeInterpreterWrapper_resizeInput()
658 input_idx, error_reporter->CachedErrorMessage()); in Java_org_tensorflow_lite_NativeInterpreterWrapper_resizeInput()
/external/mesa3d/src/broadcom/compiler/
Dnir_to_vir.c691 int8_t input_idx, uint8_t swizzle, int array_index) in emit_fragment_varying() argument
710 if (input_idx >= 0) { in emit_fragment_varying()
712 c->interp[input_idx].vp = vary; in emit_fragment_varying()
713 c->interp[input_idx].C = vir_MOV(c, r5); in emit_fragment_varying()
714 c->interp[input_idx].mode = var->data.interpolation; in emit_fragment_varying()
722 assert(input_idx < 0); in emit_fragment_varying()
782 if (input_idx >= 0) in emit_fragment_varying()
783 c->inputs[input_idx] = result; in emit_fragment_varying()
793 int input_idx = (base_attr + array_index) * 4 + chan; in emit_fragment_input() local
794 emit_fragment_varying(c, var, input_idx, chan, array_index); in emit_fragment_input()
[all …]
/external/tensorflow/tensorflow/core/example/
Dexample_parser_configuration.cc116 int input_idx = sparse_keys_start + i; in ExtractExampleParserConfiguration() local
118 op_input_tensors[input_idx].scalar<tstring>()(); in ExtractExampleParserConfiguration()
/external/tensorflow/tensorflow/lite/kernels/
Dtest_util.cc211 const int input_idx = interpreter_->inputs()[i]; in BuildInterpreter() local
212 if (input_idx == kTfLiteOptionalTensor) continue; in BuildInterpreter()
215 CHECK(interpreter_->ResizeInputTensor(input_idx, shape) == kTfLiteOk); in BuildInterpreter()
/external/tensorflow/tensorflow/lite/micro/kernels/xtensa/
Dconv.cc91 int input_idx = in ConvPerChannel() local
95 const int8_t* input_vals_offset_ptr = input_data + input_idx; in ConvPerChannel()

12