Searched refs:MakeSliceHlo (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_creation_utils.h | 50 StatusOr<HloInstruction*> MakeSliceHlo(HloInstruction* operand,
|
| D | gather_expander.cc | 136 MakeSliceHlo(index_vector, /*start_indices=*/{index_vector_dim_index}, in ExpandIndexVectorIntoOperandSpace()
|
| D | scatter_expander.cc | 159 MakeSliceHlo(index_vector, /*start_indices=*/{index_vector_dim_index}, in ExpandIndexVectorIntoOperandSpace()
|
| D | hlo_creation_utils.cc | 70 StatusOr<HloInstruction*> MakeSliceHlo(HloInstruction* operand, in MakeSliceHlo() function
|
| D | algebraic_simplifier.cc | 2294 MakeSliceHlo(nonzero_pad, start_indices, end_indices, strides)); in HandlePad()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_conv_padding_legalization.cc | 113 MakeSliceHlo(input, start_indices, limit_indices, strides).ValueOrDie(); in MaybePaddedAndSlicedInput()
|