/external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
D | operations.h | 210 BHWDC CalculateOutputShape(const BHWDC& input, 219 BHWDC CalculateOutputShape(const BHWDC& input, const Pooling3DAttributes& attr); 229 absl::Status CalculateOutputShape(const std::vector<BHWDC>& input, 231 BHWDC* output_shape); 240 Padding3D CalculateSamePadding(const BHWDC& input, 250 Padding3D CalculateSamePadding(const BHWDC& input, 278 BHWDC CalculateOutputShape(const BHWDC& input, 288 Padding3D CalculateSamePadding(const BHWDC& input, 311 Padding3D CalculateSamePadding(const BHWDC& input, 322 BHWDC CalculateOutputShape(const BHWDC& input, [all …]
|
D | operations.cc | 286 int32_t CalculateOutputWithoutStrides(const BHWDC& input, in CalculateOutputWithoutStrides() 304 int32_t CalculateOutputWithoutStrides(const BHWDC& input, in CalculateOutputWithoutStrides() 321 int32_t CalculateOutput(const BHWDC& input, in CalculateOutput() 339 int32_t CalculateOutput(const BHWDC& input, const AttrT& attr) { in CalculateOutput() 363 int32_t CalculateSamePadding(const BHWDC& input, in CalculateSamePadding() 379 int32_t CalculateSamePadding(const BHWDC& input, in CalculateSamePadding() 394 int32_t CalculateSamePadding(const BHWDC& input, in CalculateSamePadding() 408 int32_t CalculateSamePadding(const BHWDC& input, in CalculateSamePadding() 425 Padding3D MakeSamePadding(const BHWDC& input, in MakeSamePadding() 453 Padding3D MakeSamePadding(const BHWDC& input, const AttrT& attr) { in MakeSamePadding() [all …]
|
D | shape.h | 60 BHWDC = 12, enumerator 198 using BHWDC = StrongShape<Layout::BHWDC>; variable 372 TFLITE_GPU_LAYOUT_TRAITS(BHWDC, Axis::BATCH, Axis::HEIGHT, Axis::WIDTH, 617 case Layout::BHWDC: 618 return f.template operator()<Layout::BHWDC>();
|
D | tensor.h | 94 using Tensor5DFloat32 = Tensor<BHWDC, DataType::FLOAT32>;
|
D | shape.cc | 104 case Layout::BHWDC: in ToString()
|
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
D | tensor_test.cc | 61 absl::Status Tensor5DGenericTest(const BHWDC& shape, in Tensor5DGenericTest() 111 BHWDC(1, 6, 7, 4, 3), {data_type, storage_type, Layout::HWDC}, env)); in TensorTests() 113 BHWDC(1, 1, 4, 3, 12), {data_type, storage_type, Layout::HWDC}, env)); in TensorTests() 115 BHWDC(1, 6, 1, 7, 7), {data_type, storage_type, Layout::HWDC}, env)); in TensorTests() 119 BHWDC(2, 6, 7, 1, 3), {data_type, storage_type, Layout::BHWDC}, env)); in TensorTests() 121 BHWDC(4, 1, 4, 2, 12), {data_type, storage_type, Layout::BHWDC}, env)); in TensorTests() 123 BHWDC(7, 6, 1, 3, 7), {data_type, storage_type, Layout::BHWDC}, env)); in TensorTests() 125 BHWDC(13, 7, 3, 4, 3), {data_type, storage_type, Layout::BHWDC}, env)); in TensorTests() 199 BHWDC(1, 6, 14, 7, 1), in TEST_F() 203 BHWDC(1, 6, 14, 4, 2), in TEST_F() [all …]
|
D | tensor.h | 46 Tensor(cl_mem memory, bool memory_owner, const BHWDC& shape, 51 const BHWDC& shape, const TensorDescriptor& descriptor); 100 absl::Status IsValid(const BHWDC& shape) const; 114 BHWDC shape_; 124 absl::Status AllocateTensorMemory(const CLContext& context, const BHWDC& shape, 131 absl::Status CreateTensor(const CLContext& context, const BHWDC& shape, 140 const BHWDC& shape,
|
D | tensor.cc | 33 absl::Status AllocateTensorMemory(const CLContext& context, const BHWDC& shape, in AllocateTensorMemory() 216 absl::Status CreateTensor(const CLContext& context, const BHWDC& shape, in CreateTensor() 239 absl::Status CreateTensorShared(const CLContext& context, const BHWDC& shape, in CreateTensorShared() 266 Tensor::Tensor(cl_mem memory, bool memory_owner, const BHWDC& shape, in Tensor() 283 const BHWDC& shape, const TensorDescriptor& descriptor) in Tensor() 432 absl::Status Tensor::IsValid(const BHWDC& shape) const { in IsValid() 626 const BHWDC shape5D(shape.b, shape.h, shape.w, 1, shape.c); in CreateTensor() 630 absl::Status CreateTensor(const CLContext& context, const BHWDC& shape, in CreateTensor() 639 const BHWDC shape5D(shape.b, shape.h, shape.w, 1, shape.c); in CreateSharedTensor() 644 const BHWDC& shape, in CreateSharedTensor() [all …]
|
D | serialization.cc | 126 case Layout::BHWDC: in ToFB() 127 return data::Layout::BHWDC; in ToFB() 220 case data::Layout::BHWDC: in ToEnum() 221 return Layout::BHWDC; in ToEnum()
|
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
D | metal_spatial_tensor_test.mm | 37 using tflite::gpu::BHWDC; 71 static absl::Status Tensor5DGenericTest(const BHWDC& shape, 121 BHWDC(1, 6, 7, 4, 3), {data_type, storage_type, Layout::HWDC}, device)); 123 BHWDC(1, 1, 4, 3, 12), {data_type, storage_type, Layout::HWDC}, device)); 125 BHWDC(1, 6, 1, 7, 7), {data_type, storage_type, Layout::HWDC}, device)); 129 BHWDC(2, 6, 7, 1, 3), {data_type, storage_type, Layout::BHWDC}, device)); 131 BHWDC(4, 1, 4, 2, 12), {data_type, storage_type, Layout::BHWDC}, device)); 133 BHWDC(7, 6, 1, 3, 7), {data_type, storage_type, Layout::BHWDC}, device)); 135 BHWDC(13, 7, 3, 4, 3), {data_type, storage_type, Layout::BHWDC}, device));
|
D | metal_spatial_tensor.h | 43 const BHWDC& shape, const TensorDescriptor& descriptor); 89 absl::Status IsValid(const BHWDC& shape) const; 102 BHWDC shape_; 110 absl::Status CreateTensor(id<MTLDevice> device, const BHWDC& shape, 118 absl::Status CreateSharedBufferTensor(id<MTLBuffer> buffer, const BHWDC& shape,
|
D | metal_spatial_tensor.cc | 29 absl::Status CreateTextureBuffer(id<MTLBuffer> buffer, const BHWDC& shape, in CreateTextureBuffer() 57 absl::Status AllocateTensorMemory(id<MTLDevice> device, const BHWDC& shape, in AllocateTensorMemory() 153 absl::Status CreateTensor(id<MTLDevice> device, const BHWDC& shape, in CreateTensor() 187 const BHWDC& shape, in MetalSpatialTensor() 335 absl::Status MetalSpatialTensor::IsValid(const BHWDC& shape) const { in IsValid() 548 const BHWDC shape5D(shape.b, shape.h, shape.w, 1, shape.c); in CreateTensor() 552 absl::Status CreateTensor(id<MTLDevice> device, const BHWDC& shape, in CreateTensor() 561 const BHWDC shape5D(shape.b, shape.h, shape.w, 1, shape.c); in CreateSharedBufferTensor() 572 absl::Status CreateSharedBufferTensor(id<MTLBuffer> buffer, const BHWDC& shape, in CreateSharedBufferTensor()
|
/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
D | tensor_desc.cc | 721 (IsBatchedWidth() && layout == Layout::BHWDC)) { in GetPhysicalCoords() 723 } else if (layout == Layout::BHWDC) { in GetPhysicalCoords() 759 (layout == Layout::BHWC || layout == Layout::BHWDC)) { in HasAxis() 763 (layout == Layout::HWDC || layout == Layout::BHWDC)) { in HasAxis() 769 int TensorDescriptor::GetWidthSize(BHWDC shape) const { in GetWidthSize() 786 int TensorDescriptor::GetSliceStrideSize(BHWDC shape) const { in GetSliceStrideSize() 888 shape = BHWDC(src.shape.b, src.shape.h, src.shape.w, 1, src.shape.c); in UploadData() 894 shape = BHWDC(1, src.shape.h, src.shape.w, 1, src.shape.c); in UploadData() 900 shape = BHWDC(1, 1, 1, 1, src.shape.v); in UploadData() 960 int GetLinearIndex(const TensorDescriptor& desc, const BHWDC& shape, int b, in GetLinearIndex() [all …]
|
D | tensor_desc.h | 74 int GetWidthSize(BHWDC shape) const; 75 int GetSliceStrideSize(BHWDC shape) const; 103 BHWDC shape; 190 void DataFromBHWDC(const float* src, const BHWDC& shape, 194 void DataToBHWDC(const T* src, const BHWDC& shape, const TensorDescriptor& desc,
|
D | serialization_base.fbs | 138 BHWDC = 4, 141 table BHWDC { 154 shape:BHWDC;
|
D | storage_type_util.cc | 26 bool CanCreateTensorWithShape(const GpuInfo& gpu_info, const BHWDC& shape, in CanCreateTensorWithShape() 74 const BHWDC shape5D(shape.b, shape.h, shape.w, 1, shape.c); in CanCreateTensorWithShape()
|
D | storage_type_util.h | 28 bool CanCreateTensorWithShape(const GpuInfo& gpu_info, const BHWDC& shape,
|
D | serialization_base_generated.h | 60 struct BHWDC; 275 BHWDC = 4, enumerator 277 MAX = BHWDC 286 Layout::BHWDC in EnumValuesLayout() 304 if (flatbuffers::IsOutRange(e, Layout::UNKNOWN, Layout::BHWDC)) return ""; in EnumNameLayout() 1303 struct BHWDC FLATBUFFERS_FINAL_CLASS : private flatbuffers::Table { 1339 typedef BHWDC Table; 1343 fbb_.AddElement<int32_t>(BHWDC::VT_B, b, 0); in add_b() 1346 fbb_.AddElement<int32_t>(BHWDC::VT_H, h, 0); in add_h() 1349 fbb_.AddElement<int32_t>(BHWDC::VT_W, w, 0); in add_w() [all …]
|
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
D | conv_powervr.h | 122 const BHWDC* dst_shape = nullptr); 164 const BHWDC* dst_shape); 186 const BHWDC* dst_shape = nullptr); 383 const BHWDC* dst_shape = nullptr);
|
D | reduce.h | 63 const BHWDC& src_shape, OperationType op_type,
|
D | reduce.cc | 102 const BHWDC& shape) { in GetSizesFromShape() 483 const BHWDC& src_shape, OperationType op_type, in CreateReduce()
|
D | conv_powervr.cc | 219 const GpuInfo& gpu_info, const BHWDC* dst_shape) in ConvPowerVR() 1358 const Convolution3DAttributes& attr, const BHWDC* dst_shape) { in GuessBestParams() 1479 const BHWDC* dst_shape) { in CreateConvPowerVR3D()
|