Home
last modified time | relevance | path

Searched defs:num_input (Results 1 – 22 of 22) sorted by relevance

/aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_2/
Dlsh_projection_float16.mod.py17 num_input = 3 variable
Dlsh_projection_3_relaxed.mod.py17 num_input = 3 variable
Dlsh_projection_4_relaxed.mod.py17 num_input = 3 variable
Dlsh_projection_deprecated.mod.py17 num_input = 3 variable
/aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_0/
Dlsh_projection_2.mod.py17 num_input = 3 variable
Dlsh_projection_weights_as_inputs.mod.py17 num_input = 3 variable
Dlsh_projection.mod.py17 num_input = 3 variable
/aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_1/
Dlsh_projection_weights_as_inputs_relaxed.mod.py17 num_input = 3 variable
Dlsh_projection_relaxed.mod.py17 num_input = 3 variable
Dlsh_projection_2_relaxed.mod.py17 num_input = 3 variable
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Didentify_lstm_split_inputs.cc67 int num_input = model->GetArray(curr_op->inputs[LstmCellOperator::DATA_INPUT]) in Run() local
H A Didentify_lstm_merge_inputs.cc65 int num_input = model->GetArray(src_op->inputs[kInputToInputWeightsTensor]) in Run() local
/aosp_15_r20/external/ComputeLibrary/tests/validation/reference/
H A DQLSTMLayerNormalization.cpp58 const uint32_t num_input = src.shape()[0]; in qlstm_layer_normalization() local
/aosp_15_r20/external/mesa3d/src/gallium/auxiliary/tgsi/
H A Dtgsi_aa_point.c48 unsigned num_input; // number of inputs member
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/
H A DNEQLSTMLayerNormalizationKernel.cpp47 … std::pair<int64_t, int64_t> compute_mean_variance(int64_t sum, int64_t sum_sq, uint32_t num_input) in compute_mean_variance()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DDepthwiseConv3d.cu126 const int num_input = grad_input.size(0) * grad_input.stride(0); in conv_depthwise3d_cuda_backward_input_kernel() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
H A Doptimize.cc1558 int num_input = input_ty.getNumElements(); in ComputeSliceK() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Ddtensor_device.cc787 void VerifyTensorRankAndDType(TFE_TensorHandle** tensors, int num_input, in VerifyTensorRankAndDType()
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Dencapsulate_tpu_computations_pass.cc1867 int num_input = n->num_inputs(); in MakeIdentityNodesForArgsToLift() local
/aosp_15_r20/external/webrtc/modules/audio_processing/
H A Daudio_processing_unittest.cc2100 int num_input; in TEST_P() member
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dconvert_nodes_test.cc3651 const int64_t num_input = DimsAdapter(dims_array).Volume(); in TEST_P() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Doptimized_ops.h1028 int num_input; in HybridConv() local