Home
last modified time | relevance | path

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

123

/external/ComputeLibrary/tests/validation/reference/
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
/external/tensorflow/tensorflow/core/kernels/
Dstateless_random_ops_v2_util.h29 Status GetScalar(const Tensor& tensor, int input_idx, T* result) { in GetScalar()
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()
Daggregate_ops.cc58 for (int input_idx = 0; input_idx < num; ++input_idx) { in Compute() local
/external/tensorflow/tensorflow/core/kernels/linalg/
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
/external/tensorflow/tensorflow/core/framework/
Dfull_type_inference_util.cc260 FullTypeId t, int input_idx, in ContainerMap()
297 ForwardTypeInferenceFn MapCovariant(FullTypeId t, FullTypeId u, int input_idx) { in MapCovariant()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dkernel_thunk.cc74 int input_idx = 0; in PrintBufferContents() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dreduce_decomposer.cc135 Shape ExpectedOutputShape(HloReduceInstruction* reduce, int input_idx) { in ExpectedOutputShape()
/external/tensorflow/tensorflow/lite/kernels/shim/test_op/
Dsimple_tflite_op_test.cc40 std::vector<int> input_idx; in SimpleOpModel() local
/external/tensorflow/tensorflow/lite/tools/optimize/
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
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
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
/external/ComputeLibrary/src/graph/mutators/
DGroupedConvolutionMutator.cpp52 …const unsigned int input_idx = get_dimension_idx(input_tensor_desc.layout, DataLayoutD… in create_grouped_convolution() local
/external/tensorflow/tensorflow/core/grappler/
Dgraph_topology_view.cc125 const int input_idx = it->second; in InitializeFromGraph() local
/external/tensorflow/tensorflow/core/example/
Dexample_parser_configuration.cc116 int input_idx = sparse_keys_start + i; in ExtractExampleParserConfiguration() local
/external/mesa3d/src/broadcom/compiler/
Dnir_to_vir.c691 int8_t input_idx, uint8_t swizzle, int array_index) in emit_fragment_varying()
793 int input_idx = (base_attr + array_index) * 4 + chan; in emit_fragment_input() local
807 int input_idx = (attr + loc_offset) * 4 + chan; in emit_compact_fragment_input() local
1745 unsigned input_idx = c->num_inputs++; in ntq_setup_gs_inputs() local
2729 const uint32_t input_idx = in ntq_emit_intrinsic() local
/external/tensorflow/tensorflow/lite/delegates/hexagon/
Dhexagon_delegate_kernel.cc99 for (int input_idx = 0; input_idx < node->inputs->size; ++input_idx) { in Eval() local
/external/boringssl/src/crypto/cipher_extra/
Dtls_cbc.c217 size_t input_idx = 0; in EVP_sha1_final_with_secret_suffix() local
/external/rust/crates/quiche/deps/boringssl/src/crypto/cipher_extra/
Dtls_cbc.c217 size_t input_idx = 0; in EVP_sha1_final_with_secret_suffix() local
/external/cronet/third_party/boringssl/src/crypto/cipher_extra/
Dtls_cbc.c217 size_t input_idx = 0; in EVP_sha1_final_with_secret_suffix() local
/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/tools/
Dtflite_op_coverage_spec_getters_gen.cc182 int input_idx = 0; in GenerateStaticQuantOp() local
/external/tensorflow/tensorflow/core/common_runtime/
Dplacer_inspection_required_ops_utils.cc164 Status AddInputIdentity(Node* node, int input_idx, Graph* graph, in AddInputIdentity()
/external/tensorflow/tensorflow/lite/kernels/parse_example/
Dparse_example_test.cc256 const int input_idx = interpreter_->inputs()[i]; in ResizeInputTensor() local
/external/tensorflow/tensorflow/lite/kernels/
Dtest_util.cc247 const int input_idx = interpreter_->inputs()[i]; in BuildInterpreter() local
/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/lite/
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

123