Home
last modified time | relevance | path

Searched refs:MakeSliceHlo (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_creation_utils.h50 StatusOr<HloInstruction*> MakeSliceHlo(HloInstruction* operand,
Dgather_expander.cc136 MakeSliceHlo(index_vector, /*start_indices=*/{index_vector_dim_index}, in ExpandIndexVectorIntoOperandSpace()
Dscatter_expander.cc159 MakeSliceHlo(index_vector, /*start_indices=*/{index_vector_dim_index}, in ExpandIndexVectorIntoOperandSpace()
Dhlo_creation_utils.cc70 StatusOr<HloInstruction*> MakeSliceHlo(HloInstruction* operand, in MakeSliceHlo() function
Dalgebraic_simplifier.cc2294 MakeSliceHlo(nonzero_pad, start_indices, end_indices, strides)); in HandlePad()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_padding_legalization.cc113 MakeSliceHlo(input, start_indices, limit_indices, strides).ValueOrDie(); in MaybePaddedAndSlicedInput()