Home
last modified time | relevance | path

Searched refs:CreateScalarMaxComputation (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Darithmetic.cc67 XlaComputation CreateScalarMaxComputation(PrimitiveType type, in CreateScalarMaxComputation() function
187 reducer = CreateScalarMaxComputation(input_shape.element_type(), builder); in ArgMinMaxTwoPass()
213 CreateScalarMaxComputation(output_type, builder), in ArgMinMaxTwoPass()
Darithmetic.h47 XlaComputation CreateScalarMaxComputation(PrimitiveType type,
Dpooling.cc141 auto max_computation = CreateScalarMaxComputation(dtype, b); in MaxPool()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dreduce_test.cc576 auto max = CreateScalarMaxComputation(F32, &builder); in XLA_TEST_F()
588 auto max = CreateScalarMaxComputation(F32, &builder); in XLA_TEST_F()
631 auto max = CreateScalarMaxComputation(U32, &builder); in XLA_TEST_F()
775 static_cast<FuncGeneratorForType>(CreateScalarMaxComputation), in XLA_TEST_F()
865 XlaComputation max_f32 = CreateScalarMaxComputation(F32, &builder); in XLA_TEST_F()
893 XlaComputation max_fn = CreateScalarMaxComputation( in DoTest()
Dreduce_window_test.cc90 CreateScalarMaxComputation(FloatType(), &builder_), in ReduceWindowMax()
653 : CreateScalarMaxComputation(FloatType(), &b); in DoIt()
1054 : CreateScalarMaxComputation(FloatType(), &b); in XLA_TEST_P()
1261 : CreateScalarMaxComputation(FloatType(), &b); in DoIt()
1462 : CreateScalarMaxComputation(FloatType(), &b); in XLA_TEST_P()
Dselect_and_scatter_test.cc59 max_f32_ = CreateScalarMaxComputation(F32, &builder_); in SelectAndScatterTest()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtopk_ops.cc168 CreateScalarMaxComputation(xla::F32, builder), {1}); in CreateTopKUnique()
179 CreateScalarMaxComputation(xla::S32, builder), {1}); in CreateTopKUnique()