Home
last modified time | relevance | path

Searched refs:hwc_tensor (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Dadd.cc48 const auto* hwc_tensor = in GenerateCode() local
51 if (hwc_tensor) { in GenerateCode()
53 const std::string x_coord = hwc_tensor->shape.w == 1 ? "0" : "gid.x"; in GenerateCode()
54 const std::string y_coord = hwc_tensor->shape.h == 1 ? "0" : "gid.y"; in GenerateCode()
55 const std::string s_coord = hwc_tensor->shape.c == 1 ? "0" : "gid.z"; in GenerateCode()
58 if (hwc_tensor->shape.c == 1) { in GenerateCode()
69 uint3(hwc_tensor->shape.w, hwc_tensor->shape.h, in GenerateCode()
70 DivideRoundUp(hwc_tensor->shape.c, 4)), in GenerateCode()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dprelu_test_util.cc66 ::tflite::gpu::Tensor<HWC, DataType::FLOAT32> hwc_tensor; in PReLUHWCAlphaTest()
67 hwc_tensor.shape = HWC(2, 1, 2); in PReLUHWCAlphaTest()
68 hwc_tensor.data = {0.5f, -2.0f, 0.7f, 4.7f}; in PReLUHWCAlphaTest()
69 attr.alpha = hwc_tensor; in PReLUHWCAlphaTest()
Delementwise.cc312 const auto* hwc_tensor = in CreateElementwiseDesc() local
322 } else if (hwc_tensor) { in CreateElementwiseDesc()
323 return CreateElementwiseTwoInput(gpu_info, definition, op_type, *hwc_tensor, in CreateElementwiseDesc()
Delementwise_test_util.cc748 ::tflite::gpu::Tensor<HWC, DataType::FLOAT32> hwc_tensor; in MaximumWithConstantHWCTensorTest()
749 hwc_tensor.shape = HWC(2, 1, 2); in MaximumWithConstantHWCTensorTest()
750 hwc_tensor.data = {0.5f, 2.0f, 0.7f, 4.7f}; in MaximumWithConstantHWCTensorTest()
752 attr.param = hwc_tensor; in MaximumWithConstantHWCTensorTest()
780 ::tflite::gpu::Tensor<HWC, DataType::FLOAT32> hwc_tensor; in MaximumWithConstantHWCTensorBroadcastChannelsTest()
781 hwc_tensor.shape = HWC(2, 1, 1); in MaximumWithConstantHWCTensorBroadcastChannelsTest()
782 hwc_tensor.data = {0.5f, 2.0f}; in MaximumWithConstantHWCTensorBroadcastChannelsTest()
784 attr.param = hwc_tensor; in MaximumWithConstantHWCTensorBroadcastChannelsTest()