| /external/XNNPACK/test/ |
| D | subgraph-unary-tester.h | 44 num_output_elements = batch_size * channels; in UnaryTest()
49 input = std::vector<InputType>(num_output_elements + XNN_EXTRA_BYTES / sizeof(InputType)); in UnaryTest()
50 operator_output = std::vector<OutputType>(num_output_elements); in UnaryTest()
51 subgraph_output = std::vector<OutputType>(num_output_elements); in UnaryTest()
82 size_t num_output_elements; variable
|
| D | hardswish.cc | 61 std::vector<float> input(num_output_elements + XNN_EXTRA_BYTES / sizeof(float), std::nanf("")); in TEST_F()
64 std::vector<float> subgraph_output(num_output_elements, std::nanf("")); in TEST_F()
77 std::vector<float> operator_output(num_output_elements, std::nanf("")); in TEST_F()
113 for (size_t i = 0; i < num_output_elements; i++) { in TEST_F()
|
| D | binary-elementwise-operator-tester.h | 228 const size_t num_output_elements = in TestQS8() local
247 std::vector<int8_t> output(num_output_elements); in TestQS8()
248 std::vector<float> output_ref(num_output_elements); in TestQS8()
410 const size_t num_output_elements = in TestQU8() local
429 std::vector<uint8_t> output(num_output_elements); in TestQU8()
430 std::vector<float> output_ref(num_output_elements); in TestQU8()
585 const size_t num_output_elements = in TestF16() local
604 std::vector<uint16_t> output(num_output_elements); in TestF16()
605 std::vector<float> output_ref(num_output_elements); in TestF16()
806 const size_t num_output_elements = in TestF32() local
[all …]
|
| D | constant-pad-operator-tester.h | 87 inline size_t num_output_elements() const { in num_output_elements() function
141 std::vector<uint8_t> output(num_output_elements()); in TestX8()
142 std::vector<uint8_t> output_ref(num_output_elements()); in TestX8()
256 std::vector<uint16_t> output(num_output_elements()); in TestX16()
257 std::vector<uint16_t> output_ref(num_output_elements()); in TestX16()
371 std::vector<uint32_t> output(num_output_elements()); in TestX32()
372 std::vector<uint32_t> output_ref(num_output_elements()); in TestX32()
|
| D | depth-to-space.cc | 43 size_t num_output_elements = NumElements(output_dims); in DepthToSpaceTest() local
45 operator_output = std::vector<T>(num_output_elements); in DepthToSpaceTest()
46 subgraph_output = std::vector<T>(num_output_elements); in DepthToSpaceTest()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | reshape.cc | 61 int64_t non_zero_num_output_elements = 1, num_output_elements = 1; in ResizeOutput() local
72 num_output_elements *= value; in ResizeOutput()
76 if (num_input_elements == 0 && num_output_elements != 0) { in ResizeOutput()
82 num_output_elements *= output_shape->data[stretch_dim]; in ResizeOutput()
85 TF_LITE_ENSURE_EQ(context, num_input_elements, num_output_elements); in ResizeOutput()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | bcast.h | 44 int64_t num_output_elements = 1; in ComputeBatchIndices() local
52 for (int64_t k = 0; k < (dim - 1) * num_output_elements; ++k) { in ComputeBatchIndices()
53 (*out_indices)[num_output_elements + k] = (*out_indices)[k] + incr; in ComputeBatchIndices()
55 num_output_elements *= dim; in ComputeBatchIndices()
|
| /external/XNNPACK/src/subgraph/ |
| D | static-reshape.c | 186 const size_t num_output_elements = xnn_shape_multiply_all_dims(&output_value->shape); in xnn_define_static_reshape() local
188 if (num_input_elements != num_output_elements) { in xnn_define_static_reshape()
193 num_output_elements); in xnn_define_static_reshape()
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | UpSampleBicubic2d.cu | 186 const int num_output_elements = output_height * output_width; in upsample_bicubic2d_out_cuda_template() local
208 <<<ceil_div(num_output_elements, max_threads), in upsample_bicubic2d_out_cuda_template()
212 num_output_elements, in upsample_bicubic2d_out_cuda_template()
|
| D | ReduceMomentKernel.cu | 43 factor_t factor = static_cast<factor_t>(iter.num_output_elements()) / iter.numel(); in mean_kernel_impl()
|
| D | Reduce.cuh | 1002 int64_t num_outputs = iter.num_output_elements(); in setReduceConfig()
|
| /external/pytorch/aten/src/ATen/ |
| D | TensorIterator.h | 295 int64_t num_output_elements() const;
|
| D | TensorIterator.cpp | 742 int64_t TensorIteratorBase::num_output_elements() const { in num_output_elements() function in at::TensorIteratorBase
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 1049 const int num_output_elements = output_type.getNumElements(); in verify() local
1051 if (num_output_elements % z_out != 0) { in verify()
1057 if (z_in != 0 && num_input_elements / z_in != num_output_elements / z_out) { in verify()
|
| /external/pytorch/aten/src/ATen/native/ |
| D | ReduceOps.cpp | 1821 …t64_t reducing_over_num_elements = iter.num_output_elements() == 0 ? 0 : iter.numel() / iter.num_o… in warn_invalid_degrees_of_freedom()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/ |
| D | ops.mlir | 2484 // expected-error @+1 {{num_input_elements / z_in != num_output_elements / z_out}}
|