Home
last modified time | relevance | path

Searched refs:MakeConcatHlo (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_creation_utils.h118 StatusOr<HloInstruction*> MakeConcatHlo(
Dgather_expander.cc145 return MakeConcatHlo(expanded_index_components, /*dimension=*/0); in ExpandIndexVectorIntoOperandSpace()
Dscatter_expander.cc168 return MakeConcatHlo(expanded_index_components, /*dimension=*/0); in ExpandIndexVectorIntoOperandSpace()
Dspace_to_batch_converter.cc450 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()
Dhlo_creation_utils.cc233 StatusOr<HloInstruction*> MakeConcatHlo( in MakeConcatHlo() function
Dalgebraic_simplifier.cc1037 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/
Dhorizontal_loop_fusion.cc395 TF_ASSIGN_OR_RETURN(auto concated_output, MakeConcatHlo(reshapes, 0)); in CreateFusedComputation()