Searched refs:hwc_tensor (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | prelu_test_util.cc | 98 ::tflite::gpu::Tensor<HWC, DataType::FLOAT32> hwc_tensor; in PReLUHWCAlphaTest()
99 hwc_tensor.shape = HWC(2, 1, 2); in PReLUHWCAlphaTest()
100 hwc_tensor.data = {0.5f, -2.0f, 0.7f, 4.7f}; in PReLUHWCAlphaTest()
101 attr.alpha = hwc_tensor; in PReLUHWCAlphaTest()
|
| D | elementwise.cc | 285 const auto* hwc_tensor = in CreateElementwise() local
295 } else if (hwc_tensor) { in CreateElementwise()
296 return CreateElementwiseTwoInput(gpu_info, definition, op_type, *hwc_tensor, in CreateElementwise()
|
| D | elementwise_test_util.cc | 650 ::tflite::gpu::Tensor<HWC, DataType::FLOAT32> hwc_tensor; in MaximumWithConstantHWCTensorTest()
651 hwc_tensor.shape = HWC(2, 1, 2); in MaximumWithConstantHWCTensorTest()
652 hwc_tensor.data = {0.5f, 2.0f, 0.7f, 4.7f}; in MaximumWithConstantHWCTensorTest()
654 attr.param = hwc_tensor; in MaximumWithConstantHWCTensorTest()
682 ::tflite::gpu::Tensor<HWC, DataType::FLOAT32> hwc_tensor; in MaximumWithConstantHWCTensorBroadcastChannelsTest()
683 hwc_tensor.shape = HWC(2, 1, 1); in MaximumWithConstantHWCTensorBroadcastChannelsTest()
684 hwc_tensor.data = {0.5f, 2.0f}; in MaximumWithConstantHWCTensorBroadcastChannelsTest()
686 attr.param = hwc_tensor; in MaximumWithConstantHWCTensorBroadcastChannelsTest()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/ |
| D | add.cc | 44 const auto* hwc_tensor = in GenerateCode() local
46 if (hwc_tensor) { in GenerateCode()
|