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
Dhlo_verifier.cc887 Shape compare_shape = compare->root_instruction()->shape(); in HandleSort() local
Ddynamic_padder.cc1360 Shape compare_shape = in RewriteDynamicSort() local
Dalgebraic_simplifier.cc2657 Shape compare_shape = in MinMaxToClamp() local
3794 Shape compare_shape = ShapeUtil::ChangeElementType(a->shape(), PRED); in TryRemainderToAnd() local
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc2369 const Shape& compare_shape = in HandleDynamicUpdateSlice() local