Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Doperations.h219 BHWC CalculateOutputShape(const BHWC& input,
224 BHWDC CalculateOutputShape(const BHWDC& input,
229 BHWC CalculateOutputShape(const BHWC& input, const Pooling2DAttributes& attr);
233 BHWDC CalculateOutputShape(const BHWDC& input, const Pooling3DAttributes& attr);
237 absl::Status CalculateOutputShape(const std::vector<BHWC>& input,
243 absl::Status CalculateOutputShape(const std::vector<BHWDC>& input,
314 BHWC CalculateOutputShape(const BHWC& input,
319 BHWDC CalculateOutputShape(const BHWDC& input,
357 BHWC CalculateOutputShape(const BHWC& input,
363 BHWDC CalculateOutputShape(const BHWDC& input,
[all …]
Doperations.cc507 BHWC CalculateOutputShape(const BHWC& input, in CalculateOutputShape() function
517 BHWDC CalculateOutputShape(const BHWDC& input, in CalculateOutputShape() function
529 BHWC CalculateOutputShape(const BHWC& input, const Pooling2DAttributes& attr) { in CalculateOutputShape() function
534 BHWDC CalculateOutputShape(const BHWDC& input, in CalculateOutputShape() function
541 BHWC CalculateOutputShape(const BHWC& input, in CalculateOutputShape() function
548 BHWDC CalculateOutputShape(const BHWDC& input, in CalculateOutputShape() function
556 BHWC CalculateOutputShape(const BHWC& input, in CalculateOutputShape() function
563 BHWDC CalculateOutputShape(const BHWDC& input, in CalculateOutputShape() function
571 BHWC CalculateOutputShape(const BHWC& input, in CalculateOutputShape() function
579 BHWDC CalculateOutputShape(const BHWDC& input, in CalculateOutputShape() function
[all …]
Dmodel_builder.cc2768 output_value->tensor.shape = CalculateOutputShape(input_shape, attr); in Parse()
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/kernels/
Dconv_test.mm69 auto dst_shape = CalculateOutputShape(src_shape, attr);