Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Darithmetic.cc88 XlaComputation CreateScalarMinComputation(PrimitiveType type, in CreateScalarMinComputation() function
136 reducer = CreateScalarMinComputation(input_shape.element_type(), builder); in ArgMinMax()
Darithmetic.h44 XlaComputation CreateScalarMinComputation(PrimitiveType type,
/external/tensorflow/tensorflow/compiler/xla/tests/
Dreduce_test.cc602 auto min = CreateScalarMinComputation(F32, &builder); in XLA_TEST_F()
618 auto min = CreateScalarMinComputation(U32, &builder); in XLA_TEST_F()
784 static_cast<FuncGeneratorForType>(CreateScalarMinComputation), in XLA_TEST_F()
Dselect_and_scatter_test.cc60 min_f32_ = CreateScalarMinComputation(F32, &builder_); in SelectAndScatterTest()
Dreduce_window_test.cc101 CreateScalarMinComputation(FloatType(), &builder_), in ReduceWindowMin()