Searched refs:frontend_attributes (Results 1 – 10 of 10) sorted by relevance
108 xla::FrontendAttributes merged_attributes = b->frontend_attributes(); in Compute()
437 string frontend_attributes; in FunctionalizeLoop() local439 &frontend_attributes) in FunctionalizeLoop()441 builder.Attr(kXlaFrontendAttributesAttrName, frontend_attributes); in FunctionalizeLoop()
1477 void set_frontend_attributes(FrontendAttributes frontend_attributes) { in set_frontend_attributes() argument1478 frontend_attributes_ = std::move(frontend_attributes); in set_frontend_attributes()1481 const FrontendAttributes& frontend_attributes() const { in frontend_attributes() function2026 const FrontendAttributes& frontend_attributes);
280 bool ParseFrontendAttributes(FrontendAttributes* frontend_attributes);691 optional<FrontendAttributes> frontend_attributes; in ParseInstructionRhs() local694 /*required=*/false, AttrTy::kFrontendAttributes, &frontend_attributes}; in ParseInstructionRhs()1905 FrontendAttributes* frontend_attributes) { in ParseFrontendAttributes() argument1906 CHECK(frontend_attributes != nullptr); in ParseFrontendAttributes()1922 (*frontend_attributes->mutable_map())[attribute] = lexer_.GetStrVal(); in ParseFrontendAttributes()2857 FrontendAttributes frontend_attributes; in ParseAttributeHelper() local2858 if (!ParseFrontendAttributes(&frontend_attributes)) { in ParseAttributeHelper()2862 ->emplace(frontend_attributes); in ParseAttributeHelper()
701 instruction->set_frontend_attributes(proto.frontend_attributes()); in CreateFromProto()1235 broadcast->set_frontend_attributes(operand->frontend_attributes()); in CreateBroadcastSequence()1261 reshaped_operand->set_frontend_attributes(operand->frontend_attributes()); in CreateBroadcastSequence()1269 broadcast->set_frontend_attributes(operand->frontend_attributes()); in CreateBroadcastSequence()3316 const FrontendAttributes& frontend_attributes) { in FrontendAttributesToString() argument3318 frontend_attributes.map().begin(), frontend_attributes.map().end()); in FrontendAttributesToString()
898 if (new_instruction->frontend_attributes().map().empty()) { in ReplaceInstruction()900 old_instruction->frontend_attributes()); in ReplaceInstruction()
239 xla.FrontendAttributes frontend_attributes = 68; field
168 void SetFrontendAttributes(const FrontendAttributes& frontend_attributes) { in SetFrontendAttributes() argument169 frontend_attributes_ = frontend_attributes; in SetFrontendAttributes()176 const FrontendAttributes& frontend_attributes) { in SwapFrontendAttributes() argument178 frontend_attributes_ = frontend_attributes; in SwapFrontendAttributes()183 const FrontendAttributes& frontend_attributes() const { in frontend_attributes() function
1004 ExpectAttributesMatch(inst->frontend_attributes(), *expected_it); in ExpectInstructionsAttributesMatch()
339 auto* frontend_attributes = instr_proto->mutable_frontend_attributes(); in SetInstructionFrontendAttribute() local340 (*frontend_attributes->mutable_map())[attribute] = std::move(value); in SetInstructionFrontendAttribute()