Home
last modified time | relevance | path

Searched refs:weight_type (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dlstm_test.cc41 bool use_projection_bias, const TensorType weight_type, in LSTMOpModel() argument
47 weight_type_(weight_type) { in LSTMOpModel()
53 input_to_input_weights_ = AddInput({weight_type, {n_cell, n_input}}); in LSTMOpModel()
55 input_to_forget_weights_ = AddInput({weight_type, {n_cell, n_input}}); in LSTMOpModel()
56 input_to_cell_weights_ = AddInput({weight_type, {n_cell, n_input}}); in LSTMOpModel()
57 input_to_output_weights_ = AddInput({weight_type, {n_cell, n_input}}); in LSTMOpModel()
62 recurrent_to_input_weights_ = AddInput({weight_type, {n_cell, n_output}}); in LSTMOpModel()
64 recurrent_to_forget_weights_ = AddInput({weight_type, {n_cell, n_output}}); in LSTMOpModel()
65 recurrent_to_cell_weights_ = AddInput({weight_type, {n_cell, n_output}}); in LSTMOpModel()
66 recurrent_to_output_weights_ = AddInput({weight_type, {n_cell, n_output}}); in LSTMOpModel()
[all …]
Dbidirectional_sequence_lstm_test.cc49 const auto weight_type = in BidirectionalLSTMOpModel() local
55 fw_input_to_input_weights_ = AddInput(weight_type); in BidirectionalLSTMOpModel()
58 fw_input_to_forget_weights_ = AddInput(weight_type); in BidirectionalLSTMOpModel()
59 fw_input_to_cell_weights_ = AddInput(weight_type); in BidirectionalLSTMOpModel()
60 fw_input_to_output_weights_ = AddInput(weight_type); in BidirectionalLSTMOpModel()
65 fw_recurrent_to_input_weights_ = AddInput(weight_type); in BidirectionalLSTMOpModel()
68 fw_recurrent_to_forget_weights_ = AddInput(weight_type); in BidirectionalLSTMOpModel()
69 fw_recurrent_to_cell_weights_ = AddInput(weight_type); in BidirectionalLSTMOpModel()
70 fw_recurrent_to_output_weights_ = AddInput(weight_type); in BidirectionalLSTMOpModel()
76 fw_cell_to_input_weights_ = AddInput(weight_type); in BidirectionalLSTMOpModel()
[all …]
Dembedding_lookup_test.cc43 TensorType weight_type = TensorType_FLOAT32, in BaseEmbeddingLookupOpModel() argument
46 weight_ = AddInput(weight_type); in BaseEmbeddingLookupOpModel()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dlstm_full_test.cc37 bool use_projection_bias, const TensorType weight_type, in LSTMOpModel() argument
64 weight_type_(weight_type) { in LSTMOpModel()
70 AddConstInput({weight_type, {n_cell, n_input}}, input_to_input_weights); in LSTMOpModel()
72 AddConstInput({weight_type, {n_cell, n_input}}, input_to_forget_weights); in LSTMOpModel()
73 AddConstInput({weight_type, {n_cell, n_input}}, input_to_cell_weights); in LSTMOpModel()
74 AddConstInput({weight_type, {n_cell, n_input}}, input_to_output_weights); in LSTMOpModel()
79 AddConstInput({weight_type, {n_cell, n_output}}, in LSTMOpModel()
82 AddConstInput({weight_type, {n_cell, n_output}}, in LSTMOpModel()
84 AddConstInput({weight_type, {n_cell, n_output}}, recurrent_to_cell_weights); in LSTMOpModel()
85 AddConstInput({weight_type, {n_cell, n_output}}, in LSTMOpModel()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/estimators/
Darithmetic_count_util.h54 auto weight_type = in GetArithmeticCountForConvAndFullyconnectedOp() local
56 if (weight_type == nullptr || !weight_type.hasStaticShape()) return false; in GetArithmeticCountForConvAndFullyconnectedOp()
67 const int64_t cost_per_col = 2 * weight_type.getNumElements(); in GetArithmeticCountForConvAndFullyconnectedOp()
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_utils.cc81 RankedTensorType weight_type, in buildRescaleOpConvOutput() argument
94 weight_type.getElementType() in buildRescaleOpConvOutput()
117 weight_type.getElementType() in buildRescaleOpConvOutput()
Dlegalize_utils.h56 RankedTensorType weight_type,
/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dlstm_utils_test.cc52 auto weight_type = RankedTensorType::get(weight_shape, builder->getF32Type()); in createLstmCompositeFunc() local
59 SmallVector<mlir::Type, 4> input_types{input_type, weight_type, bias_type, in createLstmCompositeFunc()
Dlstm_utils.cc672 RankedTensorType weight_type = in ConvertKerasLSTMLayer() local
674 if (weight_type.getRank() != 2) in ConvertKerasLSTMLayer()
678 Transpose2D(builder, weight_kernel, weight_type, func_op.getLoc()); in ConvertKerasLSTMLayer()
/external/tensorflow/tensorflow/lite/experimental/kernels/
Dunidirectional_sequence_gru_test.cc36 const TensorType& weight_type = TensorType_FLOAT32) in GRUOpModel() argument
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc3096 const TensorType weight_type) in LSTMOpModel() argument
3101 weight_type_(weight_type) { in LSTMOpModel()
3107 input_to_input_weights_ = AddInput(weight_type); in LSTMOpModel()
3110 input_to_forget_weights_ = AddInput(weight_type); in LSTMOpModel()
3111 input_to_cell_weights_ = AddInput(weight_type); in LSTMOpModel()
3112 input_to_output_weights_ = AddInput(weight_type); in LSTMOpModel()
3117 recurrent_to_input_weights_ = AddInput(weight_type); in LSTMOpModel()
3120 recurrent_to_forget_weights_ = AddInput(weight_type); in LSTMOpModel()
3121 recurrent_to_cell_weights_ = AddInput(weight_type); in LSTMOpModel()
3122 recurrent_to_output_weights_ = AddInput(weight_type); in LSTMOpModel()
[all …]
Dnnapi_delegate.cc2338 const TfLiteType weight_type = in Validate() local
2342 Expect(weight_type == kTfLiteUInt8, in Validate()
2387 Expect(weight_type == kTfLiteFloat32 || weight_type == kTfLiteUInt8 || in Validate()
2388 weight_type == kTfLiteInt8, in Validate()
2392 Expect(weight_type == kTfLiteFloat32 || weight_type == kTfLiteUInt8, in Validate()
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_utils_v1.py676 def standardize_sample_or_class_weights(x_weight, output_names, weight_type): argument
703 raise ValueError('Provided `' + weight_type + '` was a list of ' +
706 'You should provide one `' + weight_type + '`'
710 generic_utils.check_for_unexpected_keys(weight_type, x_weight, output_names)
716 raise TypeError('The model has multiple outputs, so `' + weight_type + '` '
718 'Provided `' + weight_type + '` type not understood: ' +