Searched defs:weight_type (Results 1 – 12 of 12) sorted by relevance
54 auto weight_type = in GetArithmeticCountForConvAndFullyconnectedOp() local
36 const TensorType& weight_type = TensorType_FLOAT32) in GRUOpModel()
41 bool use_projection_bias, const TensorType weight_type, in LSTMOpModel()419 TensorType weight_type; in TEST_P() local496 TensorType weight_type; in TEST_P() local566 TensorType weight_type; in TEST_P() local1149 TensorType weight_type; in TEST_P() local1244 TensorType weight_type; in TEST_P() local2129 HybridSparseLSTMOpModel( in HybridSparseLSTMOpModel()
43 TensorType weight_type = TensorType_FLOAT32, in BaseEmbeddingLookupOpModel()
49 const auto weight_type = in BidirectionalLSTMOpModel() local
52 auto weight_type = RankedTensorType::get(weight_shape, builder->getF32Type()); in createLstmCompositeFunc() local
672 RankedTensorType weight_type = in ConvertKerasLSTMLayer() local
37 bool use_projection_bias, const TensorType weight_type, in LSTMOpModel()
81 RankedTensorType weight_type, in buildRescaleOpConvOutput()
676 def standardize_sample_or_class_weights(x_weight, output_names, weight_type): argument
3096 const TensorType weight_type) in LSTMOpModel()4754 TensorType weight_type = TensorType_FLOAT32) { in BaseEmbeddingLookupOpModel()
2338 const TfLiteType weight_type = in Validate() local