Searched refs:CalculateOutputShape (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
D | operations.h | 219 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 …]
|
D | operations.cc | 507 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 …]
|
D | model_builder.cc | 2768 output_value->tensor.shape = CalculateOutputShape(input_shape, attr); in Parse()
|
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/kernels/ |
D | conv_test.mm | 69 auto dst_shape = CalculateOutputShape(src_shape, attr);
|