| /external/tensorflow/tensorflow/lite/kernels/ |
| D | unidirectional_sequence_lstm.cc | 221 const TfLiteTensor* projection_weights = in CheckInputTensorDimensions() local
223 if (projection_weights != nullptr) { in CheckInputTensorDimensions()
224 TF_LITE_ENSURE_EQ(context, projection_weights->dims->size, 2); in CheckInputTensorDimensions()
225 TF_LITE_ENSURE_EQ(context, projection_weights->dims->data[0], n_output); in CheckInputTensorDimensions()
226 TF_LITE_ENSURE_EQ(context, projection_weights->dims->data[1], n_cell); in CheckInputTensorDimensions()
242 ((projection_weights != nullptr) || (projection_bias == nullptr)); in CheckInputTensorDimensions()
477 const TfLiteTensor* projection_weights = in Eval() local
515 forget_gate_bias, cell_bias, output_gate_bias, projection_weights, in Eval()
547 forget_gate_bias, cell_bias, output_gate_bias, projection_weights, in Eval()
|
| D | lstm_eval.h | 49 const TfLiteTensor* projection_weights, const TfLiteTensor* projection_bias,
78 const TfLiteTensor* projection_weights, const TfLiteTensor* projection_bias,
|
| D | lstm.cc | 238 const TfLiteTensor* projection_weights = in CheckInputTensorDimensions() local
240 if (projection_weights != nullptr) { in CheckInputTensorDimensions()
241 TF_LITE_ENSURE_EQ(context, projection_weights->dims->size, 2); in CheckInputTensorDimensions()
242 TF_LITE_ENSURE_EQ(context, projection_weights->dims->data[0], n_output); in CheckInputTensorDimensions()
243 TF_LITE_ENSURE_EQ(context, projection_weights->dims->data[1], n_cell); in CheckInputTensorDimensions()
259 ((projection_weights != nullptr) || (projection_bias == nullptr)); in CheckInputTensorDimensions()
562 const TfLiteTensor* projection_weights = in Eval() local
593 forget_gate_bias, cell_bias, output_gate_bias, projection_weights, in Eval()
624 forget_gate_bias, cell_bias, output_gate_bias, projection_weights, in Eval()
|
| D | bidirectional_sequence_lstm.cc | 319 const TfLiteTensor* projection_weights = in CheckLstmTensorDimensionsAndTypes() local
321 if (projection_weights != nullptr) { in CheckLstmTensorDimensionsAndTypes()
322 TF_LITE_ENSURE_EQ(context, projection_weights->dims->size, 2); in CheckLstmTensorDimensionsAndTypes()
323 TF_LITE_ENSURE_EQ(context, projection_weights->dims->data[0], n_output); in CheckLstmTensorDimensionsAndTypes()
324 TF_LITE_ENSURE_EQ(context, projection_weights->dims->data[1], n_cell); in CheckLstmTensorDimensionsAndTypes()
325 TF_LITE_ENSURE_EQ(context, projection_weights->type, in CheckLstmTensorDimensionsAndTypes()
343 ((projection_weights != nullptr) || (projection_bias == nullptr)); in CheckLstmTensorDimensionsAndTypes()
|
| D | lstm_eval.cc | 893 const TfLiteTensor* projection_weights, const TfLiteTensor* projection_bias, in EvalFloat() argument
960 (projection_weights == nullptr) ? nullptr : projection_weights->data.f; in EvalFloat()
1091 const TfLiteTensor* projection_weights, const TfLiteTensor* projection_bias, in EvalHybrid() argument
1186 (projection_weights == nullptr) in EvalHybrid()
1188 : GetInt8DataPtr(projection_weights, is_uint8_hybrid); in EvalHybrid()
1190 (projection_weights == nullptr) ? 1.0f : projection_weights->params.scale; in EvalHybrid()
|
| /external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
| D | factorization_ops.py | 745 projection_weights=None): argument
767 if projection_weights is None:
768 projection_weights = 1
773 row_weights=projection_weights)[0]
778 projection_weights=None): argument
800 if projection_weights is None:
801 projection_weights = 1
806 row_weights=projection_weights)[0]
|
| D | wals.py | 319 projection_weights = features.get(
325 projection_weights=projection_weights,
331 projection_weights=projection_weights,
|
| D | wals_test.py | 83 project_row=None, projection_weights=None, argument
168 if mode == model_fn.ModeKeys.INFER and projection_weights is not None:
170 projection_weights,
293 projection_weights=[[0.2, 0.5]])[0]
310 projection_weights=[[0.6, 0.4, 0.2]])[0]
|
| D | factorization_ops_test.py | 219 projection_weights=[0.2, 0.5])
292 projection_weights=[0.6, 0.4, 0.2])
394 projection_weights=[0.5, 0.2])
471 projection_weights=[0.4, 0.7])
|
| /external/tensorflow/tensorflow/contrib/rnn/python/ops/ |
| D | rnn_cell.py | 1540 self.projection_weights = self.add_variable(
1616 new_m = math_ops.matmul(new_m, self.projection_weights)
|