Searched refs:CreateScalarMinComputation (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | arithmetic.cc | 73 XlaComputation CreateScalarMinComputation(PrimitiveType type, in CreateScalarMinComputation() function
184 reducer = CreateScalarMinComputation(input_shape.element_type(), builder); in ArgMinMaxTwoPass()
204 CreateScalarMinComputation(output_type, builder), in ArgMinMaxTwoPass()
|
| D | arithmetic.h | 51 XlaComputation CreateScalarMinComputation(PrimitiveType type,
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | reduce_test.cc | 603 auto min = CreateScalarMinComputation(F32, &builder); in XLA_TEST_F()
619 auto min = CreateScalarMinComputation(U32, &builder); in XLA_TEST_F()
785 static_cast<FuncGeneratorForType>(CreateScalarMinComputation), in XLA_TEST_F()
|
| D | select_and_scatter_test.cc | 60 min_f32_ = CreateScalarMinComputation(F32, &builder_); in SelectAndScatterTest()
|
| D | reduce_window_test.cc | 101 CreateScalarMinComputation(FloatType(), &builder_), in ReduceWindowMin()
|