Home
last modified time | relevance | path

Searched refs:input_weights (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dbasic_rnn.cc65 const TfLiteTensor* input_weights = GetInput(context, node, kWeightsTensor); in Prepare() local
75 const int num_units = input_weights->dims->data[0]; in Prepare()
77 input_weights->dims->data[1]); in Prepare()
78 TF_LITE_ENSURE_EQ(context, input_weights->dims->data[0], bias->dims->data[0]); in Prepare()
84 TF_LITE_ENSURE_EQ(context, input_weights->type, recurrent_weights->type); in Prepare()
99 input->type == kTfLiteFloat32 && (input_weights->type == kTfLiteUInt8 || in Prepare()
100 input_weights->type == kTfLiteInt8); in Prepare()
110 input_quantized->type = input_weights->type; in Prepare()
120 hidden_state_quantized->type = input_weights->type; in Prepare()
147 const TfLiteTensor* input_weights, in EvalFloat() argument
[all …]
Dunidirectional_sequence_rnn.cc60 const TfLiteTensor* input_weights = GetInput(context, node, kWeightsTensor); in Prepare() local
75 const int num_units = input_weights->dims->data[0]; in Prepare()
77 input_weights->dims->data[1]); in Prepare()
78 TF_LITE_ENSURE_EQ(context, input_weights->dims->data[0], bias->dims->data[0]); in Prepare()
84 TF_LITE_ENSURE_EQ(context, input_weights->type, recurrent_weights->type); in Prepare()
100 input->type == kTfLiteFloat32 && (input_weights->type == kTfLiteUInt8 || in Prepare()
101 input_weights->type == kTfLiteInt8); in Prepare()
111 input_quantized->type = input_weights->type; in Prepare()
121 hidden_state_quantized->type = input_weights->type; in Prepare()
147 const TfLiteTensor* input_weights, in EvalFloat() argument
[all …]
/external/libopus/src/
Dmlp.c85 sum += layer->input_weights[j*stride + i]*input[j]; in compute_dense()
113 sum += gru->input_weights[j*stride + i]*input[j]; in compute_gru()
123 sum += gru->input_weights[N + j*stride + i]*input[j]; in compute_gru()
133 sum += gru->input_weights[2*N + j*stride + i]*input[j]; in compute_gru()
Dmlp.h38 const opus_int8 *input_weights; member
46 const opus_int8 *input_weights; member
/external/tensorflow/tensorflow/contrib/factorization/kernels/
Dwals_solver_ops.cc76 const Tensor& input_weights = context->input(3); in Compute() local
90 OP_REQUIRES(context, TensorShapeUtils::IsVector(input_weights.shape()), in Compute()
111 ((input_weights.dim_size(0) > 0 && in Compute()
114 (input_weights.dim_size(0) == 0 && factor_weights.dim_size(0) == 0 && in Compute()
127 const auto& input_weights_vec = input_weights.vec<float>(); in Compute()
/external/tensorflow/tensorflow/contrib/tensor_forest/python/
Dtensor_forest.py588 input_weights=None): argument
610 if input_weights is None:
611 input_weights = []
637 input_weights,
648 input_weights,
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dquantize.cc219 const auto& input_weights = model->GetArray(op.inputs[weights_input_index]); in ChooseQuantizationForOperatorInput() local
221 !input_weights.quantization_params) { in ChooseQuantizationForOperatorInput()
228 const auto input_weights_scale = input_weights.quantization_params->scale; in ChooseQuantizationForOperatorInput()
/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/
Dmodel_ops.cc337 const Tensor& input_weights = context->input(3); in Compute() local
353 TensorInputTarget target(input_labels, input_weights, num_targets); in Compute()
Dstats_ops.cc246 const Tensor& input_weights = context->input(7); in Compute() local
304 TensorInputTarget target(input_labels, input_weights, num_targets); in Compute()
/external/tensorflow/tensorflow/contrib/tensor_forest/client/
Drandom_forest.py225 features, labels, input_weights=weights,