Home
last modified time | relevance | path

Searched refs:fusion_id (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dcontraction_fusion.cc102 Identifier fusion_id = Identifier::get("fusion", ctx); in matchAndRewrite() local
106 return attr.first == fusion_id; in matchAndRewrite()
110 NamedAttribute fusion_attr(fusion_id, ArrayAttr::get(ctx, {fusion_name})); in matchAndRewrite()
122 attrs.push_back(NamedAttribute(fusion_id, ArrayAttr::get(ctx, updated))); in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_instruction.cc360 const int64 fusion_id = proto.called_computation_ids(0); in CreateFromProto() local
362 tensorflow::gtl::FindPtrOrNull(computation_map, fusion_id); in CreateFromProto()
364 << "No fusion computation with id " << fusion_id; in CreateFromProto()