Home
last modified time | relevance | path

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

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