Searched refs:CreateScalarMaxComputation (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | arithmetic.cc | 79 XlaComputation CreateScalarMaxComputation(PrimitiveType type, in CreateScalarMaxComputation() function
139 reducer = CreateScalarMaxComputation(input_shape.element_type(), builder); in ArgMinMax()
171 CreateScalarMaxComputation(output_type, builder), in ArgMinMax()
|
| D | arithmetic.h | 40 XlaComputation CreateScalarMaxComputation(PrimitiveType type,
|
| D | pooling.cc | 140 auto max_computation = CreateScalarMaxComputation(dtype, b); in MaxPool()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | reduce_test.cc | 575 auto max = CreateScalarMaxComputation(F32, &builder); in XLA_TEST_F()
587 auto max = CreateScalarMaxComputation(F32, &builder); in XLA_TEST_F()
630 auto max = CreateScalarMaxComputation(U32, &builder); in XLA_TEST_F()
774 static_cast<FuncGeneratorForType>(CreateScalarMaxComputation), in XLA_TEST_F()
864 XlaComputation max_f32 = CreateScalarMaxComputation(F32, &builder); in XLA_TEST_F()
892 XlaComputation max_fn = CreateScalarMaxComputation( in DoTest()
|
| D | reduce_window_test.cc | 90 CreateScalarMaxComputation(FloatType(), &builder_), in ReduceWindowMax()
637 : CreateScalarMaxComputation(FloatType(), &b); in DoIt()
1038 : CreateScalarMaxComputation(FloatType(), &b); in TEST_P()
1175 : CreateScalarMaxComputation(FloatType(), &b); in DoIt()
1368 : CreateScalarMaxComputation(FloatType(), &b); in TEST_P()
|
| D | select_and_scatter_test.cc | 59 max_f32_ = CreateScalarMaxComputation(F32, &builder_); in SelectAndScatterTest()
|