Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/kernels/
Dlinalg_ops_common.cc176 for (int input_idx = 0; input_idx < context->num_inputs(); ++input_idx) { in PrepareOutputs() local
177 unused_inputs.insert(input_idx); in PrepareOutputs()
200 for (int input_idx : unused_inputs) { in PrepareOutputs() local
202 input_idx, output_idx, output_tensor_shape, &out)) { in PrepareOutputs()
204 unused_inputs.erase(input_idx); in PrepareOutputs()
Daggregate_ops.cc62 for (int input_idx = 0; input_idx < num; ++input_idx) { in Compute() local
63 if (ctx->forward_input_to_output_with_shape(input_idx, 0, input0.shape(), in Compute()
65 reused_input = input_idx; in Compute()
Ddepthtospace_op_gpu.cu.cc70 CUDA_1D_KERNEL_LOOP(input_idx, nthreads) { in D2S_NCHW()
78 const int n_bY_bX_oC_iY = input_idx / input_width; in D2S_NCHW()
79 const int iX = input_idx - n_bY_bX_oC_iY * input_width; in D2S_NCHW()
97 *(output_ptr + output_idx) = ldg(input_ptr + input_idx); in D2S_NCHW()
Dspacetodepth_op_gpu.cu.cc69 CUDA_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()
Dmkl_conv_grad_filter_ops.cc622 size_t input_idx = 0; in MakeInputTfShape() local
623 return GetTfShape(context, input_idx); in MakeInputTfShape()
/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/lite/tools/
Dverifier.cc215 for (const int input_idx : *op->inputs()) { in VerifySubGraphConsistency()
216 if (input_idx == kOptionalTensor) continue; in VerifySubGraphConsistency()
217 if (constant_tensors.find(input_idx) == constant_tensors.end() && in VerifySubGraphConsistency()
218 variable_tensors.find(input_idx) == variable_tensors.end() && in VerifySubGraphConsistency()
219 subgraph_input_tensors.find(input_idx) == in VerifySubGraphConsistency()
221 output_tensors.find(input_idx) == output_tensors.end()) { in VerifySubGraphConsistency()
223 "Input tensor %d to op %d (%s) is not produced", input_idx, in VerifySubGraphConsistency()
/external/tensorflow/tensorflow/compiler/jit/
Dnode_matchers.cc168 for (int input_idx = 0, e = input_matchers->size(); input_idx < e; in MatchAndExplain() local
169 input_idx++) { in MatchAndExplain()
170 if (!MatchAndExplainInput(node, input_idx, listener)) { in MatchAndExplain()
299 bool MatchAndExplainInput(const Node* node, int input_idx, in MatchAndExplainInput()
302 if (!node->input_edge(input_idx, &edge).ok()) { in MatchAndExplainInput()
304 *listener << "\ncould not find incoming edge for input " << input_idx; in MatchAndExplainInput()
311 if ((*input_matchers)[input_idx].MatchAndExplain(input, &inner_listener)) { in MatchAndExplainInput()
316 *listener << "\ninput " << input_idx << " does not match expected:\n"; in MatchAndExplainInput()
317 (*input_matchers)[input_idx].DescribeTo(listener->stream()); in MatchAndExplainInput()
/external/tensorflow/tensorflow/lite/tools/optimize/
Dsubgraph_quantizer.cc335 for (size_t input_idx = 0; input_idx < op->inputs.size(); ++input_idx) { in AsymmetricQuantizeInputsAndOutputs() local
336 auto input_tensor = subgraph_->tensors[op->inputs[input_idx]].get(); in AsymmetricQuantizeInputsAndOutputs()
337 if (IsSubgraphInput(op->inputs[input_idx]) && in AsymmetricQuantizeInputsAndOutputs()
340 AsymmetricQuantizeTensor(op_code, op->inputs[input_idx])); in AsymmetricQuantizeInputsAndOutputs()
Dsubgraph_quantizer_test.cc477 for (size_t input_idx = 0; input_idx < 2; ++input_idx) { in TEST() local
478 EXPECT_EQ(subgraph->tensors[op->inputs[input_idx]].get()->type, in TEST()
481 float_graph->tensors()->Get(op->inputs[input_idx])->quantization(); in TEST()
483 subgraph->tensors[op->inputs[input_idx]]->quantization.get(); in TEST()
/external/tensorflow/tensorflow/core/framework/
Dshape_inference.cc776 int input_idx, ShapeHandle* out) { in MakeShapeFromShapeTensorTreatScalarAsUnknownShape() argument
778 TF_RETURN_IF_ERROR(WithRankAtMost(input(input_idx), 1, &input_shape)); in MakeShapeFromShapeTensorTreatScalarAsUnknownShape()
780 requested_input_tensor_as_partial_shape_[input_idx] = true; in MakeShapeFromShapeTensorTreatScalarAsUnknownShape()
781 if (input_idx < input_tensors_as_shapes_.size() && in MakeShapeFromShapeTensorTreatScalarAsUnknownShape()
782 input_tensors_as_shapes_[input_idx].IsSet() && in MakeShapeFromShapeTensorTreatScalarAsUnknownShape()
783 RankKnown(input_tensors_as_shapes_[input_idx])) { in MakeShapeFromShapeTensorTreatScalarAsUnknownShape()
784 *out = input_tensors_as_shapes_[input_idx]; in MakeShapeFromShapeTensorTreatScalarAsUnknownShape()
790 input_tensor(input_idx), input_shape, out); in MakeShapeFromShapeTensorTreatScalarAsUnknownShape()
793 Status InferenceContext::MakeShapeFromShapeTensor(int input_idx, in MakeShapeFromShapeTensor() argument
796 TF_RETURN_IF_ERROR(WithRank(input(input_idx), 1, &input_shape)); in MakeShapeFromShapeTensor()
[all …]
Dshape_inference.h479 Status MakeShapeFromShapeTensor(int input_idx, ShapeHandle* out);
484 Status MakeShapeFromShapeTensorTreatScalarAsUnknownShape(int input_idx,
/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/kernels/
Dtest_util.cc128 const int input_idx = interpreter_->inputs()[i]; in BuildInterpreter() local
129 if (input_idx == kOptionalTensor) continue; in BuildInterpreter()
132 CHECK(interpreter_->ResizeInputTensor(input_idx, shape) == kTfLiteOk); in BuildInterpreter()
/external/tensorflow/tensorflow/core/example/
Dexample_parser_configuration.cc116 int input_idx = sparse_keys_start + i; in ExtractExampleParserConfiguration() local
117 (*var_len_features)[i].key = op_input_tensors[input_idx].scalar<string>()(); in ExtractExampleParserConfiguration()
/external/tensorflow/tensorflow/lite/java/src/main/native/
Dnativeinterpreterwrapper_jni.cc432 jint input_idx, jintArray dims) { in Java_org_tensorflow_lite_NativeInterpreterWrapper_resizeInput() argument
439 const int idx = static_cast<int>(input_idx); in Java_org_tensorflow_lite_NativeInterpreterWrapper_resizeInput()
448 TfLiteTensor* target = interpreter->tensor(input_idx); in Java_org_tensorflow_lite_NativeInterpreterWrapper_resizeInput()
Dnativeinterpreterwrapper_jni.h224 jint input_idx, jintArray dims);
/external/tensorflow/tensorflow/core/graph/
Dmkl_layout_pass.cc1360 int* input_idx, int list_length,
1381 int* input_idx, int list_length,
1580 const gtl::InlinedVector<std::pair<Node*, int>, 4>& inputs, int* input_idx, in GetNodesProducingTFTensorList() argument
1582 CHECK_LT(*input_idx, inputs.size()); in GetNodesProducingTFTensorList()
1589 CHECK_LT(*input_idx, inputs.size()); in GetNodesProducingTFTensorList()
1590 Node* n = inputs[*input_idx].first; in GetNodesProducingTFTensorList()
1591 int slot = inputs[*input_idx].second; in GetNodesProducingTFTensorList()
1595 (*input_idx)++; in GetNodesProducingTFTensorList()
1645 const gtl::InlinedVector<std::pair<Node*, int>, 4>& inputs, int* input_idx, in GetNodesProducingMklTensorList() argument
1647 CHECK_LT(*input_idx, inputs.size()); in GetNodesProducingMklTensorList()
[all …]
/external/mesa3d/src/gallium/drivers/radeonsi/
Dsi_shader_nir.c654 unsigned input_idx = variable->data.driver_location; in si_nir_build_llvm() local
670 declare_nir_input_fs(ctx, variable, input_idx / 4, data); in si_nir_build_llvm()
673 ctx->inputs[input_idx + chan] = in si_nir_build_llvm()
/external/tensorflow/tensorflow/python/ops/
Darray_grad.py750 input_idx = array_ops.reshape(math_ops.range(1, input_indices_num,
754 input_idx,
813 input_idx = array_ops.reshape(
817 input_idx, op.get_attr("ksizes"), op.get_attr("strides"),
/external/u-boot/drivers/net/
Dsmc91111.c890 int input_idx; in smc_read_phy_register() local
935 input_idx = clk_idx; in smc_read_phy_register()
982 if (bits[input_idx++] & MII_MDI) in smc_read_phy_register()
/external/v8/src/wasm/baseline/
Dliftoff-compiler.cc228 uint32_t input_idx = input_idx_start; in CollectReservedRegsForParameters() local
236 descriptor_->GetInputLocation(input_idx + pair_idx); in CollectReservedRegsForParameters()
248 input_idx += num_lowered_params; in CollectReservedRegsForParameters()
253 uint32_t ProcessParameter(ValueType type, uint32_t input_idx) { in ProcessParameter() argument
261 descriptor_->GetInputLocation(input_idx + pair_idx); in ProcessParameter()
341 uint32_t input_idx = kInstanceParameterIndex + 1; in StartFunctionBody() local
343 input_idx += ProcessParameter(__ local_type(param_idx), input_idx); in StartFunctionBody()
345 DCHECK_EQ(input_idx, descriptor_->InputCount()); in StartFunctionBody()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dfusion_utils.cc243 auto input_idx = std::distance(second_inputs.begin(), input_it); in FuseReturns() local
245 set_input(first_inputs, second_inputs, first_outputs, input_idx); in FuseReturns()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dfunction_optimizer.cc1509 const int input_idx = input_placeholder_idx->second; in PlaceInlinedFunctionBody() local
1511 ctx->graph_view().GetRegularFanin({&func_node, input_idx}); in PlaceInlinedFunctionBody()
1769 const int input_idx = input_placeholder_idx->second; in InlineIndirectFunctionCall() local
1770 func_body_node.add_input(inputs[input_idx].ToString()); in InlineIndirectFunctionCall()
/external/tensorflow/tensorflow/core/profiler/internal/
Dtfprof_node.h346 void AddInput(const string& input, int64 output_index, int input_idx) { in AddInput() argument
347 inputs_[input_idx] = input; in AddInput()

12