Home
last modified time | relevance | path

Searched defs:compare_shape (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dcomparison_expander.cc54 auto compare_shape = signed_shape; in BitcastConvertFloatingPointToIntegral() local
85 Shape compare_shape = lhs->shape(); in ExpandInstruction() local
H A Ddynamic_padder.cc1406 Shape compare_shape = in RewriteDynamicSort() local
H A Dhlo_verifier.cc904 Shape compare_shape = compare->root_instruction()->shape(); in HandleSort() local
H A Dalgebraic_simplifier.cc2871 Shape compare_shape = in MinMaxToClamp() local
4079 Shape compare_shape = ShapeUtil::ChangeElementType(a->shape(), PRED); in TryRemainderToAnd() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dspmd_partitioner.cc3146 const Shape& compare_shape = in HandleDynamicUpdateSlice() local