Searched defs:KeepDims (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | reduce_test.cc | 276 TEST(ConstFloatMeanOpTest, KeepDims) { in TEST() argument
415 TEST(DynamicFloatMeanOpTest, KeepDims) { in TEST() argument
456 TEST(ConstUint8MeanOpTest, KeepDims) { in TEST() argument
608 TEST(DynamicUint8MeanOpTest, KeepDims) { in TEST() argument
666 TEST(ConstFloatSumOpTest, KeepDims) { in TEST() argument
725 TEST(DynamicFloatSumOpTest, KeepDims) { in TEST() argument
778 TEST(ConstUint8SumOpTest, KeepDims) { in TEST() argument
807 TEST(DynamicUint8SumOpTest, KeepDims) { in TEST() argument
854 TEST(ConstFloatProdOpTest, KeepDims) { in TEST() argument
895 TEST(DynamicFloatProdOpTest, KeepDims) { in TEST() argument
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | fully_connected_tester.h | 67 inline FullyConnectedTester& KeepDims(bool keep_dims) { in KeepDims() function
72 inline bool KeepDims() const { return keep_dims_; } in KeepDims() function
|
| 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
|
| /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 | 1465 bool ReduceTransposer::KeepDims(const utils::MutableNodeView& node) { in KeepDims() function in tensorflow::grappler::ReduceTransposer
|
| D | layout_optimizer.cc | 1994 bool KeepDims() const { return node_->attr().at("keep_dims").b(); } in KeepDims() function in tensorflow::grappler::__anon5a8ea6d90111::ReduceProcessor
|