| /aosp_15_r20/external/pytorch/torch/_numpy/ |
| H A D | _reductions_impl.py | 67 def count_nonzero(a: ArrayLike, axis: AxisLike = None, *, keepdims: KeepDims = False):
77 keepdims: KeepDims = False,
97 keepdims: KeepDims = False,
116 keepdims: KeepDims = False,
130 keepdims: KeepDims = False,
144 keepdims: KeepDims = False,
162 keepdims: KeepDims = False,
180 keepdims: KeepDims = False,
191 keepdims: KeepDims = False,
210 keepdims: KeepDims = False,
[all …]
|
| H A D | _normalizations.py | 25 KeepDims = typing.TypeVar("KeepDims") variable
|
| H A D | linalg.py | 165 def norm(x: ArrayLike, ord=None, axis=None, keepdims: KeepDims = False):
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | reduce_test.cc | 206 TEST(ConstFloatMeanOpTest, KeepDims) { in TEST() argument
357 TEST(DynamicFloatMeanOpTest, KeepDims) { in TEST() argument
398 TEST(ConstUint8MeanOpTest, KeepDims) { in TEST() argument
570 TEST(DynamicUint8MeanOpTest, KeepDims) { in TEST() argument
689 TEST(ConstFloatSumOpTest, KeepDims) { in TEST() argument
748 TEST(DynamicFloatSumOpTest, KeepDims) { in TEST() argument
823 TEST(ConstUint8SumOpTest, KeepDims) { in TEST() argument
852 TEST(DynamicUint8SumOpTest, KeepDims) { in TEST() argument
1045 TEST(ConstFloatProdOpTest, KeepDims) { in TEST() argument
1096 TEST(DynamicFloatProdOpTest, KeepDims) { in TEST() argument
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| H A D | reduce_tester.h | 59 inline ReduceTester& KeepDims(bool keep_dims) { in KeepDims() function
64 inline bool KeepDims() const { return keep_dims_; } in KeepDims() function
|
| H A D | fully_connected_tester.h | 68 inline FullyConnectedTester& KeepDims(bool keep_dims) { in KeepDims() function
73 inline bool KeepDims() const { return keep_dims_; } in KeepDims() function
|
| H A D | quantized_reduce_tester.h | 61 inline QuantizedReduceTester& KeepDims(bool keep_dims) { in KeepDims() function
66 inline bool KeepDims() const { return keep_dims_; } in KeepDims() function
|
| H A D | quantized_fully_connected_tester.h | 116 inline QuantizedFullyConnectedTester& KeepDims(bool keep_dims) { in KeepDims() function
121 inline bool KeepDims() const { return keep_dims_; } in KeepDims() function
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/NEON/ |
| H A D | ReductionOperation.cpp | 73 const auto KeepDims = framework::dataset::make("KeepDims", { true, false }); variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| H A D | propagate_fixed_sizes.cc | 533 bool KeepDims(const Operator& op) { in KeepDims() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | generic_layout_optimizer_transposer.cc | 1460 bool ReduceTransposer::KeepDims(const utils::MutableNodeView& node) { in KeepDims() function in tensorflow::grappler::ReduceTransposer
|