| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | mean_test.cc | 45 .KeepDims(false) in TEST()
66 .KeepDims(true) in TEST()
87 .KeepDims(false) in TEST()
108 .KeepDims(true) in TEST()
129 .KeepDims(false) in TEST()
150 .KeepDims(true) in TEST()
171 .KeepDims(false)
177 .KeepDims(false)
198 .KeepDims(true)
204 .KeepDims(true)
[all …]
|
| 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
72 if (KeepDims()) { in OutputShape()
|
| 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 | fully_connected_test.cc | 63 .KeepDims(true)
108 .KeepDims(true)
178 .KeepDims(true)
227 .KeepDims(true)
|
| D | fully_connected_tester.cc | 40 if (KeepDims()) { in OutputShape()
245 KeepDims()); in CreateTfLiteModel()
|
| D | reduce_tester.cc | 136 CreateReducerOptions(builder, KeepDims()); in CreateTfLiteModel()
|
| /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/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 | 1949 if (KeepDims()) { in AddLayoutTransposeToOutputs()
1957 return KeepDims() || ((IsAlongAllFourDims() || IsAlongHWC() || in IsReduceAxisSupported()
1959 !KeepDims()); in IsReduceAxisSupported()
1994 bool KeepDims() const { return node_->attr().at("keep_dims").b(); } in KeepDims() function in tensorflow::grappler::__anon5a8ea6d90111::ReduceProcessor
|
| D | generic_layout_optimizer_transposer.h | 430 bool KeepDims(const utils::MutableNodeView& node);
|
| D | generic_layout_optimizer_transposer.cc | 1465 bool ReduceTransposer::KeepDims(const utils::MutableNodeView& node) { in KeepDims() function in tensorflow::grappler::ReduceTransposer
1506 if (KeepDims(node)) { in IsReduceAxisSupported()
1560 if (KeepDims(*node)) { in TransposeNode()
|
| D | constant_folding_test.cc | 2734 attr = attr.KeepDims(true); in TEST_F()
2803 attr = attr.KeepDims(false); in TEST_F()
2806 one_axis, attr.KeepDims(false)); in TEST_F()
2808 multiple_axes, attr.KeepDims(false)); in TEST_F()
2811 one_axis, attr.KeepDims(false)); in TEST_F()
2814 variable_axis, attr.KeepDims(false)); in TEST_F()
2817 multiple_axes, attr.KeepDims(true)); in TEST_F()
|
| D | generic_layout_optimizer_transposer_test.cc | 392 auto attrs = ops::Sum::Attrs().KeepDims(true); in ReduceTransposerKeepDims()
|
| /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()
|