Home
last modified time | relevance | path

Searched refs:test_reduce_mean (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/ops/parallel_for/
Dxla_control_flow_ops_test.py112 def test_reduce_mean(self): member in PForTest
Dcontrol_flow_ops_test.py253 def test_reduce_mean(self): member in ReductionTest
/external/tensorflow/tensorflow/compiler/mlir/tosa/tests/
Dtf-to-tosa-pipeline.mlir243 // CHECK-LABEL: test_reduce_mean
249 func @test_reduce_mean(%arg0: tensor<13x21x3xf32>) -> tensor<21x3xf32> {
Dtfl-to-tosa-pipeline.mlir239 // CHECK-LABEL: test_reduce_mean
245 func @test_reduce_mean(%arg0: tensor<13x21x3xf32>) -> tensor<21x3xf32> {