Searched defs:KeepDims (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| 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 …]
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| 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
|
| 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
|
| 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
|
| 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
|
| /external/ComputeLibrary/tests/validation/NEON/ |
| D | ReductionOperation.cpp | 73 const auto KeepDims = framework::dataset::make("KeepDims", { true, false }); variable
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | propagate_fixed_sizes.cc | 533 bool KeepDims(const Operator& op) { in KeepDims() function
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer.cc | 1460 bool ReduceTransposer::KeepDims(const utils::MutableNodeView& node) { in KeepDims() function in tensorflow::grappler::ReduceTransposer
|