Searched refs:compare_shape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | comparison_expander.cc | 54 auto compare_shape = signed_shape; in BitcastConvertFloatingPointToIntegral() local
55 compare_shape.set_element_type(PRED); in BitcastConvertFloatingPointToIntegral()
57 compare_shape, signed_value, zero, ComparisonDirection::kLt)); in BitcastConvertFloatingPointToIntegral()
85 Shape compare_shape = lhs->shape(); in ExpandInstruction() local
86 PrimitiveType compare_type = compare_shape.element_type(); in ExpandInstruction()
93 compare_shape.set_element_type(compare_type); in ExpandInstruction()
95 HloInstruction::CreateConvert(compare_shape, lhs)); in ExpandInstruction()
97 HloInstruction::CreateConvert(compare_shape, rhs)); in ExpandInstruction()
105 auto signed_shape = compare_shape; in ExpandInstruction()
107 auto unsigned_shape = compare_shape; in ExpandInstruction()
|
| D | hlo_verifier.cc | 550 Shape compare_shape = compare->root_instruction()->shape(); in HandleSort() local
551 if (!ShapeUtil::Compatible(compare_shape, ShapeUtil::MakeShape(PRED, {}))) { in HandleSort()
555 StringifyShape(compare_shape)); in HandleSort()
|
| D | dynamic_padder.cc | 1259 Shape compare_shape = in RewriteDynamicSort() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 2395 const Shape& compare_shape = in HandleDynamicUpdateSlice() local
2405 compare_shape, new_indices[dim], partition_offset, in HandleDynamicUpdateSlice()
2409 compare_shape, new_indices[dim], in HandleDynamicUpdateSlice()
2419 compare_shape, HloOpcode::kAnd, offset_ge, offset_lt)); in HandleDynamicUpdateSlice()
2422 compare_shape, HloOpcode::kAnd, all_dims_within_partition, in HandleDynamicUpdateSlice()
|