| /external/tensorflow/tensorflow/lite/kernels/ |
| D | reduce_test.cc | 247 TEST(ConstFloatMeanOpTest, KeepDims) { in TEST() argument
347 TEST(DynamicFloatMeanOpTest, KeepDims) { in TEST() argument
389 TEST(ConstUint8MeanOpTest, KeepDims) { in TEST() argument
497 TEST(DynamicUint8MeanOpTest, KeepDims) { in TEST() argument
555 TEST(ConstFloatSumOpTest, KeepDims) { in TEST() argument
594 TEST(DynamicFloatSumOpTest, KeepDims) { in TEST() argument
647 TEST(ConstUint8SumOpTest, KeepDims) { in TEST() argument
676 TEST(DynamicUint8SumOpTest, KeepDims) { in TEST() argument
723 TEST(ConstFloatProdOpTest, KeepDims) { in TEST() argument
754 TEST(DynamicFloatProdOpTest, KeepDims) { in TEST() argument
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | shape_optimizer_test.cc | 41 Output e = ops::ReduceProd(s.WithOpName("e"), c, d, attrs.KeepDims(false)); in TEST_F()
42 Output f = ops::ReduceProd(s.WithOpName("f"), c, d, attrs.KeepDims(true)); in TEST_F()
104 attrs.KeepDims(false)); in TEST_F()
|
| D | layout_optimizer.cc | 1940 if (KeepDims()) { in AddLayoutTransposeToOutputs()
1948 return KeepDims() || ((IsAlongAllFourDims() || IsAlongHWC() || in IsReduceAxisSupported()
1950 !KeepDims()); in IsReduceAxisSupported()
1985 bool KeepDims() const { return node_->attr().at("keep_dims").b(); } in KeepDims() function in tensorflow::grappler::__anon9cd52bfd0111::ReduceProcessor
|
| D | generic_layout_optimizer_transposer.h | 393 bool KeepDims(const utils::MutableNodeView& node);
|
| D | generic_layout_optimizer_transposer.cc | 1207 bool ReduceTransposer::KeepDims(const utils::MutableNodeView& node) { in KeepDims() function in tensorflow::grappler::ReduceTransposer
1241 if (KeepDims(node)) { in IsReduceAxisSupported()
1279 if (KeepDims(*node)) { in TransposeNode()
|
| D | constant_folding_test.cc | 2717 attr = attr.KeepDims(true); in TEST_F()
2786 attr = attr.KeepDims(false); in TEST_F()
2789 one_axis, attr.KeepDims(false)); in TEST_F()
2791 multiple_axes, attr.KeepDims(false)); in TEST_F()
2794 one_axis, attr.KeepDims(false)); in TEST_F()
2797 variable_axis, attr.KeepDims(false)); in TEST_F()
2800 multiple_axes, attr.KeepDims(true)); in TEST_F()
|
| D | generic_layout_optimizer_transposer_test.cc | 3530 auto attrs = ops::Sum::Attrs().KeepDims(true); in TEST_F()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | quantize_nodes_test.cc | 528 dequantize_reduction_dims, Min::Attrs().KeepDims(false)); in TestRemoveRedundantQuantization()
532 dequantize_reduction_dims, Max::Attrs().KeepDims(false)); in TestRemoveRedundantQuantization()
627 dequantize_reduction_dims, Min::Attrs().KeepDims(false)); in TestRemoveRedundantQuantizationWithBiasAdd()
631 dequantize_reduction_dims, Max::Attrs().KeepDims(false)); in TestRemoveRedundantQuantizationWithBiasAdd()
753 dequantize_reduction_dims, Min::Attrs().KeepDims(false)); in TestRemoveRedundantQuantizationWithMultipleOutputs()
757 dequantize_reduction_dims, Max::Attrs().KeepDims(false)); in TestRemoveRedundantQuantizationWithMultipleOutputs()
1167 quantize_reduction_dims1, Min::Attrs().KeepDims(false)); in TestMergeDuplicateQuantizes()
1171 quantize_reduction_dims1, Max::Attrs().KeepDims(false)); in TestMergeDuplicateQuantizes()
1194 quantize_reduction_dims2, Min::Attrs().KeepDims(false)); in TestMergeDuplicateQuantizes()
1198 quantize_reduction_dims2, Max::Attrs().KeepDims(false)); in TestMergeDuplicateQuantizes()
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | nn_grad.cc | 120 auto sum = Sum(scope, grad_inputs[0], {1}, Sum::KeepDims(true)); in LogSoftmaxGrad()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | propagate_fixed_sizes.cc | 533 bool KeepDims(const Operator& op) { in KeepDims() function
564 const bool keep_dims = KeepDims(*op); in ProcessTensorFlowReductionOperator()
|