Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.h53 HloInstruction* CreateR0WithType(PrimitiveType type, NativeT value, in CreateR0WithType() function
64 return CreateR0WithType(type, -float_pad_value, b); in CreateFirstWithType()
73 return CreateR0WithType(type, float_pad_value, b); in CreateLastWithType()
Dgather_scatter_handler.cc66 min_indices.push_back(CreateR0WithType<int32>( in IndexBoundsForGatherScatterOperandPartitionedOnTrivialSliceDims()
68 max_indices.push_back(CreateR0WithType<int32>( in IndexBoundsForGatherScatterOperandPartitionedOnTrivialSliceDims()
83 CreateR0WithType<int32>(replicated_indices.base_shape().element_type(), in IndexBoundsForGatherScatterOperandPartitionedOnTrivialSliceDims()
269 CreateR0WithType(PRED, false, b), {dnums.index_vector_dim()}, in ParititonTrivialIndexedOperandDimension()
Dfft_handler.cc372 .PadWithValue(CreateR0WithType(hlo->shape().element_type(), 0, &b_)); in HandleFft()
Dspmd_partitioner.cc1781 CreateR0WithType(hlo->shape().element_type(), 0, &b_)); in HandleCustomCall()