Home
last modified time | relevance | path

Searched refs:BHWDC (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Doperations.h210 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 …]
Doperations.cc286 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 …]
Dshape.h60 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>();
Dtensor.h94 using Tensor5DFloat32 = Tensor<BHWDC, DataType::FLOAT32>;
Dshape.cc104 case Layout::BHWDC: in ToString()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dtensor_test.cc61 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 …]
Dtensor.h46 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,
Dtensor.cc33 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 …]
Dserialization.cc126 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/
Dmetal_spatial_tensor_test.mm37 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));
Dmetal_spatial_tensor.h43 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,
Dmetal_spatial_tensor.cc29 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/
Dtensor_desc.cc721 (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 …]
Dtensor_desc.h74 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,
Dserialization_base.fbs138 BHWDC = 4,
141 table BHWDC {
154 shape:BHWDC;
Dstorage_type_util.cc26 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()
Dstorage_type_util.h28 bool CanCreateTensorWithShape(const GpuInfo& gpu_info, const BHWDC& shape,
Dserialization_base_generated.h60 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/
Dconv_powervr.h122 const BHWDC* dst_shape = nullptr);
164 const BHWDC* dst_shape);
186 const BHWDC* dst_shape = nullptr);
383 const BHWDC* dst_shape = nullptr);
Dreduce.h63 const BHWDC& src_shape, OperationType op_type,
Dreduce.cc102 const BHWDC& shape) { in GetSizesFromShape()
483 const BHWDC& src_shape, OperationType op_type, in CreateReduce()
Dconv_powervr.cc219 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()