Home
last modified time | relevance | path

Searched refs:weights_type (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconvolution_transposed.h102 DataType weights_type = definition_.precision == CalculationsPrecision::F32 in UploadWeights() local
106 std::vector<uint8_t> weights_data(flt_count * SizeOf(weights_type)); in UploadWeights()
107 RearrangeWeights(weights, GetWeightsDescription(), weights_type, in UploadWeights()
112 desc.element_type = weights_type; in UploadWeights()
126 int sub_size = SizeOf(weights_type) * 4 * texture_width * texture_height; in UploadWeights()
129 desc.element_type = weights_type; in UploadWeights()
Dconvolution_transposed_3x3_thin.cc208 DataType weights_type = definition_.precision == CalculationsPrecision::F32 in UploadWeights() local
213 desc.element_type = weights_type; in UploadWeights()
219 RearrangeWeights(weights, GetWeightsDescription(), weights_type, in UploadWeights()
257 const DataType weights_type = definition.GetDataType(); in CreateConvolutionTransposed3x3ThinDynamicWeights() local
260 {weights_type, TensorStorageType::BUFFER, Layout::HWC}); in CreateConvolutionTransposed3x3ThinDynamicWeights()
Dconvolution_transposed_4x4.cc397 DataType weights_type = definition_.precision == CalculationsPrecision::F32 in UploadWeights() local
402 desc.element_type = weights_type; in UploadWeights()
412 RearrangeWeights(weights, GetWeightsDescription(), weights_type, in UploadWeights()
450 const DataType weights_type = definition.GetDataType(); in CreateConvolutionTransposed4x4DynamicWeights() local
453 {weights_type, TensorStorageType::BUFFER, Layout::HWC}); in CreateConvolutionTransposed4x4DynamicWeights()
Dconv_powervr.h258 DataType weights_type = conv_params_.weights_data_type; in UploadWeights() local
260 std::vector<uint8_t> weights_data(flt_count * SizeOf(weights_type)); in UploadWeights()
261 RearrangeWeights(weights, GetWeightsDescription(), weights_type, in UploadWeights()
266 desc.element_type = weights_type; in UploadWeights()
284 int sub_size = SizeOf(weights_type) * 4 * texture_width * texture_height; in UploadWeights()
287 desc.element_type = weights_type; in UploadWeights()
Dconv_weights_converter_test_util.cc53 DataType weights_type = DataType::FLOAT32; in ConvolutionWeightsConverterTest() local
55 std::vector<uint8_t> weights_data(flt_count * SizeOf(weights_type)); in ConvolutionWeightsConverterTest()
56 RearrangeWeights(weights, weight_desc, weights_type, in ConvolutionWeightsConverterTest()
70 int sub_size = SizeOf(weights_type) * 4 * texture_width * texture_height; in ConvolutionWeightsConverterTest()
82 flt_count * SizeOf(weights_type)); in ConvolutionWeightsConverterTest()
Dconvolution_transposed_3x3.cc392 DataType weights_type = definition_.precision == CalculationsPrecision::F32 in UploadWeights() local
397 desc.element_type = weights_type; in UploadWeights()
407 RearrangeWeights(weights, GetWeightsDescription(), weights_type, in UploadWeights()
444 const DataType weights_type = definition.GetDataType(); in CreateConvolutionTransposed3x3DynamicWeights() local
447 {weights_type, TensorStorageType::BUFFER, Layout::HWC}); in CreateConvolutionTransposed3x3DynamicWeights()
Dconv_metal.cc594 const WeightsDescription& weights_desc, const DataType& weights_type) { in ReorderWeightsForConv() argument
597 std::vector<uint8_t> result(flt_count * SizeOf(weights_type)); in ReorderWeightsForConv()
598 RearrangeWeights(weights, weights_desc, weights_type, absl::MakeSpan(result)); in ReorderWeightsForConv()
1022 auto weights_type = DeduceDataTypeFromPrecision(definition.precision); in CreateConvolutionMetal() local
1039 weights_desc.element_type = weights_type; in CreateConvolutionMetal()
1043 attr.weights, desc.GetWeightsDescription(), weights_type); in CreateConvolutionMetal()
1050 bias_desc.element_type = weights_type; in CreateConvolutionMetal()
1054 attr.bias, weights_type, in CreateConvolutionMetal()
1147 auto weights_type = DeduceDataTypeFromPrecision(definition.precision); in CreateConvolutionMetalWino4x4To6x6() local
1156 weights_desc.element_type = weights_type; in CreateConvolutionMetalWino4x4To6x6()
[all …]
Dconvolution_transposed.cc620 const DataType weights_type = definition.GetDataType(); in CreateConvolutionTransposedDynamicWeights() local
624 {weights_type, TensorStorageType::BUFFER, Layout::HWC}); in CreateConvolutionTransposedDynamicWeights()
628 {weights_type, TensorStorageType::TEXTURE_2D, Layout::HWC}); in CreateConvolutionTransposedDynamicWeights()
630 {weights_type, TensorStorageType::TEXTURE_2D, Layout::HWC}); in CreateConvolutionTransposedDynamicWeights()
632 {weights_type, TensorStorageType::TEXTURE_2D, Layout::HWC}); in CreateConvolutionTransposedDynamicWeights()
634 {weights_type, TensorStorageType::TEXTURE_2D, Layout::HWC}); in CreateConvolutionTransposedDynamicWeights()
/external/tensorflow/tensorflow/lite/kernels/
Dunidirectional_sequence_lstm_test.cc38 const TensorType& weights_type = TensorType_FLOAT32, in UnidirectionalLSTMOpModel() argument
51 input_to_input_weights_ = AddInput(weights_type); in UnidirectionalLSTMOpModel()
54 input_to_forget_weights_ = AddInput(weights_type); in UnidirectionalLSTMOpModel()
55 input_to_cell_weights_ = AddInput(weights_type); in UnidirectionalLSTMOpModel()
56 input_to_output_weights_ = AddInput(weights_type); in UnidirectionalLSTMOpModel()
61 recurrent_to_input_weights_ = AddInput(weights_type); in UnidirectionalLSTMOpModel()
64 recurrent_to_forget_weights_ = AddInput(weights_type); in UnidirectionalLSTMOpModel()
65 recurrent_to_cell_weights_ = AddInput(weights_type); in UnidirectionalLSTMOpModel()
66 recurrent_to_output_weights_ = AddInput(weights_type); in UnidirectionalLSTMOpModel()
72 cell_to_input_weights_ = AddInput(weights_type); in UnidirectionalLSTMOpModel()
[all …]
Dtranspose_conv.cc125 TfLiteType weights_type, in AllocateTemporaryTensorsIfRequired() argument
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc352 const TfLiteType weights_type = context->tensors[weights_id].type; in IsHybridOperator() local
354 IsQuantized(weights_type); in IsHybridOperator()
361 const TfLiteType weights_type = context->tensors[weights_id].type; in IsHybridOperator() local
362 return IsFloat(input_type) && IsQuantized(weights_type); in IsHybridOperator()
369 const TfLiteType weights_type = context->tensors[weights_id].type; in IsHybridOperator() local
370 return IsFloat(input_type) && IsQuantized(weights_type); in IsHybridOperator()
376 const TfLiteType weights_type = context->tensors[weights_id].type; in IsHybridOperator() local
377 return IsFloat(input_type) && IsQuantized(weights_type); in IsHybridOperator()
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc859 const auto weights_type = weights_tensor.getType().cast<ShapedType>(); in fold() local
867 !weights_type.getElementType().isF32() || in fold()
874 if (!input_type.hasStaticShape() || !weights_type.hasStaticShape() || in fold()
881 weights_type.getShape().size() != 2 || in fold()