| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| D | texture2d_test.mm | 40 std::vector<float> gpu_data;
41 XCTAssertTrue(texture.ReadData<float>(device, &gpu_data).ok());
43 XCTAssertEqual(gpu_data.size(), data.size());
44 for (int i = 0; i < gpu_data.size(); ++i) {
45 XCTAssertEqual(gpu_data[i], data[i]);
58 std::vector<half> gpu_data;
59 XCTAssertTrue(texture.ReadData<half>(device, &gpu_data).ok());
61 XCTAssertEqual(gpu_data.size(), data.size());
62 for (int i = 0; i < gpu_data.size(); ++i) {
63 XCTAssertEqual(gpu_data[i], data[i]);
|
| D | buffer_test.mm | 44 std::vector<float> gpu_data;
45 XCTAssertTrue(buffer.ReadData<float>(&gpu_data).ok());
47 XCTAssertEqual(gpu_data.size(), data.size());
48 for (int i = 0; i < gpu_data.size(); ++i) {
49 XCTAssertEqual(gpu_data[i], data[i]);
61 std::vector<half> gpu_data;
62 XCTAssertTrue(buffer.ReadData<half>(&gpu_data).ok());
64 XCTAssertEqual(gpu_data.size(), data.size());
65 for (int i = 0; i < gpu_data.size(); ++i) {
66 XCTAssertEqual(gpu_data[i], data[i]);
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | texture2d_test.cc | 39 std::vector<float> gpu_data; in TEST_F() local
40 ASSERT_OK(texture.ReadData<float>(env_.queue(), &gpu_data)); in TEST_F()
42 EXPECT_THAT(gpu_data, Pointwise(FloatNear(0.0f), data)); in TEST_F()
52 std::vector<half> gpu_data; in TEST_F() local
53 ASSERT_OK(texture.ReadData<half>(env_.queue(), &gpu_data)); in TEST_F()
55 EXPECT_THAT(gpu_data, Pointwise(FloatNear(0.0f), data)); in TEST_F()
|
| D | buffer_test.cc | 39 std::vector<float> gpu_data; in TEST_F() local
40 ASSERT_OK(buffer.ReadData<float>(env_.queue(), &gpu_data)); in TEST_F()
42 EXPECT_THAT(gpu_data, Pointwise(FloatNear(0.0f), data)); in TEST_F()
51 std::vector<half> gpu_data; in TEST_F() local
52 ASSERT_OK(buffer.ReadData<half>(env_.queue(), &gpu_data)); in TEST_F()
54 EXPECT_THAT(gpu_data, Pointwise(FloatNear(0.0f), data)); in TEST_F()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/ |
| D | depthwise_conv_plus_1x1_conv.cc | 36 std::vector<float> gpu_data; in UploadWeights() local
37 gpu_data.reserve(dw_dst_ch_aligned + dw_weights_count + conv_dst_ch_aligned + in UploadWeights()
42 gpu_data.push_back(dw_attr.bias.data[i]); in UploadWeights()
44 gpu_data.push_back(0.0f); in UploadWeights()
56 gpu_data.push_back(dw_attr.weights.data[f_index]); in UploadWeights()
58 gpu_data.push_back(0.0f); in UploadWeights()
67 gpu_data.push_back(conv_attr.bias.data[i]); in UploadWeights()
69 gpu_data.push_back(0.0f); in UploadWeights()
83 gpu_data.push_back(conv_attr.weights.data[f_index]); in UploadWeights()
85 gpu_data.push_back(0.0f); in UploadWeights()
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | tensor_linear_desc.cc | 104 float* gpu_data = reinterpret_cast<float*>(data.data()); in UploadLinearData() local
107 gpu_data[i] = src.data[i]; in UploadLinearData()
109 gpu_data[i] = 0.0f; in UploadLinearData()
114 half* gpu_data = reinterpret_cast<half*>(data.data()); in UploadLinearData() local
117 gpu_data[i] = src.data[i]; in UploadLinearData()
119 gpu_data[i] = 0.0f; in UploadLinearData()
|
| D | tensor_desc.cc | 911 float* gpu_data = reinterpret_cast<float*>(data.data()); in UploadData() local
912 DataFromBHWDC(src, shape, *this, gpu_data); in UploadData()
914 half* gpu_data = reinterpret_cast<half*>(data.data()); in UploadData() local
915 DataFromBHWDC(src, shape, *this, gpu_data); in UploadData()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | convolution_transposed_thin.h | 85 float4* gpu_data = reinterpret_cast<float4*>(desc.data.data()); in UploadData() local
86 RearrangeWeightsData(weights, absl::MakeSpan(gpu_data, flt4_count)); in UploadData()
91 gpu_data[flt4_count] = bias_value; in UploadData()
93 half4* gpu_data = reinterpret_cast<half4*>(desc.data.data()); in UploadData() local
94 RearrangeWeightsData(weights, absl::MakeSpan(gpu_data, flt4_count)); in UploadData()
99 gpu_data[flt4_count] = bias_value; in UploadData()
|
| D | conv_powervr.h | 240 float* gpu_data = reinterpret_cast<float*>(desc.data.data()); in UploadBias() local
242 gpu_data[i] = i < bias.shape.v ? bias.data[i] : 0.0f; in UploadBias()
245 half* gpu_data = reinterpret_cast<half*>(desc.data.data()); in UploadBias() local
247 gpu_data[i] = i < bias.shape.v ? bias.data[i] : 0.0f; in UploadBias()
|
| D | conv_metal.cc | 607 float* gpu_data = reinterpret_cast<float*>(result.data()); in ReorderBiasesForConv() local
609 gpu_data[i] = i < biases.shape.v ? biases.data[i] : 0.0f; in ReorderBiasesForConv()
612 half* gpu_data = reinterpret_cast<half*>(result.data()); in ReorderBiasesForConv() local
614 gpu_data[i] = i < biases.shape.v ? biases.data[i] : 0.0f; in ReorderBiasesForConv()
|