Searched defs:KeepDims (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | reduce_test.cc | 247 TEST(ConstFloatMeanOpTest, KeepDims) { in TEST() argument
345 TEST(DynamicFloatMeanOpTest, KeepDims) { in TEST() argument
387 TEST(ConstUint8MeanOpTest, KeepDims) { in TEST() argument
450 TEST(DynamicUint8MeanOpTest, KeepDims) { in TEST() argument
508 TEST(ConstFloatSumOpTest, KeepDims) { in TEST() argument
547 TEST(DynamicFloatSumOpTest, KeepDims) { in TEST() argument
600 TEST(ConstUint8SumOpTest, KeepDims) { in TEST() argument
629 TEST(DynamicUint8SumOpTest, KeepDims) { in TEST() argument
661 TEST(ConstFloatProdOpTest, KeepDims) { in TEST() argument
692 TEST(DynamicFloatProdOpTest, KeepDims) { in TEST() argument
[all …]
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | propagate_fixed_sizes.cc | 532 bool KeepDims(const Operator& op) { in KeepDims() function
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | layout_optimizer.cc | 1935 bool KeepDims() const { return node_->attr().at("keep_dims").b(); } in KeepDims() function in tensorflow::grappler::__anon49c8f8520111::ReduceProcessor
|