Searched refs:MakeConcatHlo (Results 1 – 7 of 7) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_creation_utils.h | 118 StatusOr<HloInstruction*> MakeConcatHlo(
|
D | gather_expander.cc | 145 return MakeConcatHlo(expanded_index_components, /*dimension=*/0); in ExpandIndexVectorIntoOperandSpace()
|
D | scatter_expander.cc | 168 return MakeConcatHlo(expanded_index_components, /*dimension=*/0); in ExpandIndexVectorIntoOperandSpace()
|
D | space_to_batch_converter.cc | 450 TF_ASSIGN_OR_RETURN(activations, MakeConcatHlo({first_slice, activations}, in HaloDuplicateWithSlice() 455 TF_ASSIGN_OR_RETURN(activations, MakeConcatHlo({activations, halo_region}, in HaloDuplicateWithSlice() 2502 MakeConcatHlo(absl::MakeSpan(activations_chunks), new_spatial_dimension)); in PropagateOnBackpropFilterConv()
|
D | hlo_creation_utils.cc | 233 StatusOr<HloInstruction*> MakeConcatHlo( in MakeConcatHlo() function
|
D | algebraic_simplifier.cc | 1037 MakeConcatHlo({lhs_scatter_index, rhs_scatter_index}, in HandleAdd() 1041 MakeConcatHlo({lhs_scatter_update, rhs_scatter_update}, in HandleAdd()
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | horizontal_loop_fusion.cc | 395 TF_ASSIGN_OR_RETURN(auto concated_output, MakeConcatHlo(reshapes, 0)); in CreateFusedComputation()
|