Searched defs:weight_type (Results 1 – 16 of 16) sorted by relevance
56 auto weight_type = in GetArithmeticCountForConvAndFullyconnectedOp() local
55 auto weight_type = RankedTensorType::get(weight_shape, builder->getF32Type()); in createLstmCompositeFunc() local
678 RankedTensorType weight_type = in ConvertKerasLSTMLayer() local
213 const GpuKernelArgumentInfo::Type weight_type = _settings.export_weights_to_cl_image() ? in declare_variables() local
234 …const GpuKernelArgumentInfo::Type weight_type = _settings.export_to_cl_image() ? GpuKernelArgument… in declare_variables() 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
37 bool use_projection_bias, const TensorType weight_type, in LSTMOpModel()
478 auto weight_type = weight.getType().dyn_cast_or_null<RankedTensorType>(); in matchAndRewrite() local
85 ShapedType weight_type, ShapedType output_type) { in buildRescaleOpConvOutput()
670 def standardize_sample_or_class_weights(x_weight, output_names, weight_type): argument
3173 const TensorType weight_type) in LSTMOpModel()4831 TensorType weight_type = TensorType_FLOAT32) { in BaseEmbeddingLookupOpModel()
3000 const TfLiteType weight_type = in Validate() local
3160 auto weight_type = transpose_conv.weights() in GetArithmeticCount() local