Home
last modified time | relevance | path

Searched refs:flt_count (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dweights_conversion.cc43 const uint flt_count = in RearrangeWeights() local
50 absl::MakeSpan(f32_ptr, flt_count / 4)); in RearrangeWeights()
55 absl::MakeSpan(f16_ptr, flt_count / 4)); in RearrangeWeights()
63 absl::MakeSpan(f32_ptr, flt_count / 4)); in RearrangeWeights()
68 absl::MakeSpan(f16_ptr, flt_count / 4)); in RearrangeWeights()
76 absl::MakeSpan(f32_ptr, flt_count / 4)); in RearrangeWeights()
81 absl::MakeSpan(f16_ptr, flt_count / 4)); in RearrangeWeights()
89 absl::MakeSpan(f32_ptr, flt_count / 4)); in RearrangeWeights()
94 absl::MakeSpan(f16_ptr, flt_count / 4)); in RearrangeWeights()
103 absl::MakeSpan(f32_ptr, flt_count / 4)); in RearrangeWeights()
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconv_weights_converter_test_util.cc51 const int flt_count = in ConvolutionWeightsConverterTest() local
55 std::vector<uint8_t> weights_data(flt_count * SizeOf(weights_type)); in ConvolutionWeightsConverterTest()
79 dst_tensors[0].shape = BHWC(1, 1, 1, flt_count); in ConvolutionWeightsConverterTest()
80 dst_tensors[0].data.resize(flt_count); in ConvolutionWeightsConverterTest()
82 flt_count * SizeOf(weights_type)); in ConvolutionWeightsConverterTest()
Dconvolution_transposed.h100 const int flt_count = in UploadWeights() local
106 std::vector<uint8_t> weights_data(flt_count * SizeOf(weights_type)); in UploadWeights()
Dconvolution_transposed_3x3_thin.cc205 const int flt_count = in UploadWeights() local
216 desc.size = flt_count * SizeOf(desc.element_type); in UploadWeights()
Dconvolution_transposed_4x4.cc394 const int flt_count = in UploadWeights() local
409 desc.size = flt_count * SizeOf(desc.element_type); in UploadWeights()
Dconvolution_transposed_3x3.cc389 const int flt_count = in UploadWeights() local
404 desc.size = flt_count * SizeOf(desc.element_type); in UploadWeights()
Dconv_powervr.h256 const int flt_count = in UploadWeights() local
260 std::vector<uint8_t> weights_data(flt_count * SizeOf(weights_type)); in UploadWeights()
Dconv_metal.cc595 const int flt_count = in ReorderWeightsForConv() local
597 std::vector<uint8_t> result(flt_count * SizeOf(weights_type)); in ReorderWeightsForConv()