Home
last modified time | relevance | path

Searched refs:CreateScalarComparisonComputation (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/
H A Dcomparators.cc37 XlaComputation CreateScalarComparisonComputation( in CreateScalarComparisonComputation() function
43 return CreateScalarComparisonComputation(name, operand_types, generators, in CreateScalarComparisonComputation()
48 XlaComputation CreateScalarComparisonComputation( in CreateScalarComparisonComputation() function
116 return CreateScalarComparisonComputation("compare-less-than", operand_types, in CreateScalarLtComputation()
123 return CreateScalarComparisonComputation( in CreateScalarGtComputation()
H A Dcomparators.h48 XlaComputation CreateScalarComparisonComputation(
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dunique_op.cc181 auto lt_chain = xla::CreateScalarComparisonComputation( in CompileWithAxis()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dcustom_call_handler.cc188 XlaComputation comparator = CreateScalarComparisonComputation( in HandleCustomCallTopK()