Home
last modified time | relevance | path

Searched refs:CreateScalarMinComputation (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/
H A Darithmetic.h51 XlaComputation CreateScalarMinComputation(PrimitiveType type,
H A Darithmetic.cc73 XlaComputation CreateScalarMinComputation(PrimitiveType type, in CreateScalarMinComputation() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dreduce_test.cc610 auto min = CreateScalarMinComputation(F32, &builder); in XLA_TEST_F()
626 auto min = CreateScalarMinComputation(U32, &builder); in XLA_TEST_F()
792 static_cast<FuncGeneratorForType>(CreateScalarMinComputation), in XLA_TEST_F()
H A Dselect_and_scatter_test.cc62 min_f32_ = CreateScalarMinComputation(F32, &builder_); in SelectAndScatterTest()
H A Dreduce_window_test.cc99 CreateScalarMinComputation(FloatType(), &builder_), in ReduceWindowMin()