Searched refs:CreateScalarMaxComputation (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | reduce_test.cc | 583 auto max = CreateScalarMaxComputation(F32, &builder); in XLA_TEST_F()
595 auto max = CreateScalarMaxComputation(F32, &builder); in XLA_TEST_F()
638 auto max = CreateScalarMaxComputation(U32, &builder); in XLA_TEST_F()
782 static_cast<FuncGeneratorForType>(CreateScalarMaxComputation), in XLA_TEST_F()
872 XlaComputation max_f32 = CreateScalarMaxComputation(F32, &builder); in XLA_TEST_F()
900 XlaComputation max_fn = CreateScalarMaxComputation( in DoTest()
|
| H A D | reduce_window_test.cc | 88 CreateScalarMaxComputation(FloatType(), &builder_), in ReduceWindowMax()
651 : CreateScalarMaxComputation(FloatType(), &b); in DoIt()
1079 : CreateScalarMaxComputation(FloatType(), &b); in XLA_TEST_P()
1310 : CreateScalarMaxComputation(FloatType(), &b); in DoIt()
1535 : CreateScalarMaxComputation(FloatType(), &b); in XLA_TEST_P()
|
| H A D | select_and_scatter_test.cc | 61 max_f32_ = CreateScalarMaxComputation(F32, &builder_); in SelectAndScatterTest()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | topk_ops.cc | 169 CreateScalarMaxComputation(xla::F32, builder), {1}); in CreateTopKUnique()
180 CreateScalarMaxComputation(xla::S32, builder), {1}); in CreateTopKUnique()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| H A D | arithmetic.h | 47 XlaComputation CreateScalarMaxComputation(PrimitiveType type,
|
| H A D | arithmetic.cc | 67 XlaComputation CreateScalarMaxComputation(PrimitiveType type, in CreateScalarMaxComputation() function
|
| H A D | pooling.cc | 141 auto max_computation = CreateScalarMaxComputation(dtype, b); in MaxPool()
|