Home
last modified time | relevance | path

Searched defs:input_idx (Results 1 – 25 of 71) sorted by relevance

123

/aosp_15_r20/external/pytorch/torch/autograd/
H A Dgradcheck.py466 fn, input_idx, inputs, outputs, eps, input=None, is_forward_ad=False argument
615 def _with_prepare_inputs(fn, inputs, input_idx, input_to_perturb, fast_mode=False): argument
658 fn, input_idx, inputs, u, eps, is_forward_ad=False argument
867 def _get_analytical_jacobian(inputs, outputs, input_idx, output_idx): argument
1043 output_idx, input_idx, res, exp, is_forward_ad=False argument
1376 input_idx, argument
1748 func, tupled_inputs, outputs, input_idx, output_idx, rtol, atol, eps, is_forward_ad argument
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/
H A Dquantize_model.cc240 const int32_t input_idx = input.first; in PopulateRealValueOpSet() local
375 int32_t input_idx) { in InputQuantizeRequired()
562 const int32_t input_idx = SetInputType( in SetInputAndOutputTypes() local
724 for (size_t input_idx = 0; input_idx < op->inputs.size(); ++input_idx) { in ApplyConstraints() local
862 for (const int32_t input_idx : subgraph->inputs) { in IsSubgraphInput() local
876 int32_t input_idx = input.first; in QuantizeOpInput() local
1661 const int32_t input_idx = input.first; in FillQuantizationParams() local
H A Dmodify_model_interface.cc60 for (size_t input_idx = 0; input_idx < subgraph->inputs.size(); in GetInputTensors() local
403 for (size_t input_idx = 0; input_idx < subgraph->inputs.size(); in AddUint8Dequant() local
H A Dquantize_model_test.cc294 const int32_t input_idx = subgraph->inputs[0]; in TEST_P() local
361 const int32_t input_idx = subgraph->inputs[0]; in TEST_P() local
1142 for (size_t input_idx = 0; input_idx < 2; ++input_idx) { in TEST_F() local
1212 for (size_t input_idx = 0; input_idx < 2; ++input_idx) { in TEST_P() local
/aosp_15_r20/external/ComputeLibrary/tests/validation/reference/
H A DQLSTMLayerNormalization.cpp65 for(uint32_t input_idx = 0; input_idx < num_input; ++input_idx) in qlstm_layer_normalization() local
81 for(uint32_t input_idx = 0; input_idx < num_input; ++input_idx) in qlstm_layer_normalization() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/
H A Dlinalg_ops_common.cc185 for (int input_idx = 0; input_idx < context->num_inputs(); ++input_idx) { in PrepareOutputs() local
209 for (int input_idx : unused_inputs) { in PrepareOutputs() local
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dfull_type_inference_util.cc260 FullTypeId t, int input_idx, in ContainerMap()
297 ForwardTypeInferenceFn MapCovariant(FullTypeId t, FullTypeId u, int input_idx) { in MapCovariant()
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/
H A Dscalar_type_analysis.cpp190 size_t input_idx = 0; in InferExpectedScalarType() local
311 size_t input_idx = 0; in UpdateScalarTypeForInputs() local
/aosp_15_r20/external/boringssl/src/crypto/cipher_extra/
H A Dtls_cbc.c217 size_t input_idx = 0; in EVP_sha1_final_with_secret_suffix() local
310 size_t input_idx = 0; in EVP_sha256_final_with_secret_suffix() local
/aosp_15_r20/external/cronet/third_party/boringssl/src/crypto/cipher_extra/
H A Dtls_cbc.c217 size_t input_idx = 0; in EVP_sha1_final_with_secret_suffix() local
310 size_t input_idx = 0; in EVP_sha256_final_with_secret_suffix() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dstateful_random_ops.cc155 Status GetScalar(const Tensor& tensor, int input_idx, T* result) { in GetScalar()
172 Status GetAlg(OpKernelContext* ctx, int input_idx, Algorithm* alg) { in GetAlg()
H A Dstateless_random_ops_v2_util.h29 Status GetScalar(const Tensor& tensor, int input_idx, T* result) { in GetScalar()
H A Daggregate_ops.cc58 for (int input_idx = 0; input_idx < num; ++input_idx) { in Compute() local
/aosp_15_r20/external/mesa3d/src/broadcom/compiler/
H A Dnir_to_vir.c1042 int8_t input_idx, uint8_t swizzle, int array_index) in emit_fragment_varying()
1120 int input_idx = (base_attr + array_index) * 4 + chan; in emit_fragment_input() local
1134 int input_idx = (attr + loc_offset) * 4 + chan; in emit_compact_fragment_input() local
2351 unsigned input_idx = c->num_inputs++; in ntq_setup_gs_inputs() local
2367 unsigned input_idx = c->num_inputs++; in ntq_setup_gs_inputs() local
3884 const uint32_t input_idx = in ntq_emit_intrinsic() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DDistributionTemplates.h212 int input_idx = thread_idx + base_index; in distribution_binary_elementwise_kernel() local
227 int input_idx = thread_idx + base_index; in distribution_binary_elementwise_kernel() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dkernel_thunk.cc74 int input_idx = 0; in PrintBufferContents() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/lite/
H A Dquantize_model_test.cc357 const int input_idx = 0; in TEST_F() local
841 for (size_t input_idx = 0; input_idx < 2; ++input_idx) { in TEST_F() local
910 for (size_t input_idx = 0; input_idx < 2; ++input_idx) { in TEST_P() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/shim/test_op/
H A Dsimple_tflite_op_test.cc40 std::vector<int> input_idx; in SimpleOpModel() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dreduce_decomposer.cc135 Shape ExpectedOutputShape(HloReduceInstruction* reduce, int input_idx) { in ExpectedOutputShape()
/aosp_15_r20/external/executorch/exir/backend/test/demos/rpc/
H A DExecutorBackend.cpp161 for (size_t input_idx = 0; input_idx < num_inputs; input_idx++) { in execute() local
/aosp_15_r20/external/ComputeLibrary/src/graph/mutators/
H A DGroupedConvolutionMutator.cpp52 …const unsigned int input_idx = get_dimension_idx(input_tensor_desc.layout, DataLayoutD… in create_grouped_convolution() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/
H A Dgraph_topology_view.cc125 const int input_idx = it->second; in InitializeFromGraph() local
/aosp_15_r20/external/tensorflow/tensorflow/core/example/
H A Dexample_parser_configuration.cc116 int input_idx = sparse_keys_start + i; in ExtractExampleParserConfiguration() local
/aosp_15_r20/external/rust/android-crates-io/crates/quiche/deps/boringssl/src/crypto/cipher_extra/
Dtls_cbc.c217 size_t input_idx = 0; in EVP_sha1_final_with_secret_suffix() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/
H A Dhexagon_delegate_kernel.cc99 for (int input_idx = 0; input_idx < node->inputs->size; ++input_idx) { in Eval() local

123