Searched refs:MakeSliceHlo (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_creation_utils.h | 55 StatusOr<HloInstruction*> MakeSliceHlo(HloInstruction* operand,
|
| D | space_to_batch_converter.cc | 396 TF_ASSIGN_OR_RETURN(first_slice, MakeSliceHlo(activations, start_indices, in HaloDuplicateWithSlice()
418 MakeSliceHlo(activations, start_indices_halo, in HaloDuplicateWithSlice()
445 MakeSliceHlo(activations, start_indices_activations_cut, in HaloDuplicateWithSlice()
1511 MakeSliceHlo(new_consumer, start_indices, end_indices, strides)); in Propagate()
1522 MakeSliceHlo(new_consumer, start_indices_top, in Propagate()
1617 MakeSliceHlo(new_consumer, start_indices, end_indices, strides)); in Propagate()
1748 MakeSliceHlo(reshape, start_indices, end_indices, strides)); in BatchToSpace()
|
| 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 | 81 StatusOr<HloInstruction*> MakeSliceHlo(HloInstruction* operand, in MakeSliceHlo() function
|
| D | algebraic_simplifier.cc | 3508 MakeSliceHlo(nonzero_pad, start_indices, end_indices, strides)); in HandlePad()
4193 MakeSliceHlo(pad_operand, new_starts, new_limits, in HandleSlice()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | horizontal_loop_fusion.cc | 412 MakeSliceHlo(concated_output, {slice_start}, {slice_limit}, in CreateFusedComputation()
|
| D | gpu_conv_padding_legalization.cc | 113 MakeSliceHlo(input, start_indices, limit_indices, strides).ValueOrDie(); in MaybePaddedAndSlicedInput()
|