Home
last modified time | relevance | path

Searched refs:frontend_attributes (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dlegalize-tf-communication.mlir25 …// CHECK-SAME: mhlo.frontend_attributes = {_xla_host_transfer_handler_name = "tf_rendezvous", _xla…
32 …// CHECK-SAME: mhlo.frontend_attributes = {_xla_host_transfer_handler_name = "tf_rendezvous", _xla…
41 …// CHECK-SAME: mhlo.frontend_attributes = {_xla_host_transfer_handler_name = "tf_rendezvous", _xla…
48 …// CHECK-SAME: mhlo.frontend_attributes = {_xla_host_transfer_handler_name = "tf_rendezvous", _xla…
124 …// CHECK-SAME: mhlo.frontend_attributes = {_xla_host_transfer_handler_name = "tf_rendezvous", _xla…
142 …// CHECK-SAME: mhlo.frontend_attributes = {_xla_host_transfer_handler_name = "tf_rendezvous", _xla…
162 …// CHECK-SAME: mhlo.frontend_attributes = {_xla_host_transfer_handler_name = "tf_rendezvous", _xla…
166 …// CHECK-SAME: mhlo.frontend_attributes = {_xla_host_transfer_handler_name = "tf_rendezvous", _xla…
172 …// CHECK-SAME: mhlo.frontend_attributes = {_xla_host_transfer_handler_name = "tf_rendezvous", _xla…
176 …// CHECK-SAME: mhlo.frontend_attributes = {_xla_host_transfer_handler_name = "tf_rendezvous", _xla…
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dtriangular_solve_rewriter.cc62 custom_call->set_frontend_attributes(instr->frontend_attributes()); in Run()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compilation_device.cc126 xla::FrontendAttributes merged_attributes = b->frontend_attributes(); in Compute()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_instruction.h1751 void set_frontend_attributes(FrontendAttributes frontend_attributes) { in set_frontend_attributes() argument
1752 frontend_attributes_ = std::move(frontend_attributes); in set_frontend_attributes()
1755 void add_frontend_attributes(FrontendAttributes frontend_attributes) { in add_frontend_attributes() argument
1757 frontend_attributes.map().begin(), frontend_attributes.map().end()); in add_frontend_attributes()
1760 const FrontendAttributes& frontend_attributes() const { in frontend_attributes() function
2412 const FrontendAttributes& frontend_attributes);
Dhlo_parser.cc450 bool ParseFrontendAttributes(FrontendAttributes* frontend_attributes);
1153 optional<FrontendAttributes> frontend_attributes; in ParseInstructionRhs() local
1156 /*required=*/false, AttrTy::kFrontendAttributes, &frontend_attributes}; in ParseInstructionRhs()
1239 if (frontend_attributes) { in ParseInstructionRhs()
1240 instruction->set_frontend_attributes(*frontend_attributes); in ParseInstructionRhs()
3002 FrontendAttributes* frontend_attributes) { in ParseFrontendAttributes() argument
3003 CHECK(frontend_attributes != nullptr); in ParseFrontendAttributes()
3019 (*frontend_attributes->mutable_map())[attribute] = lexer_.GetStrVal(); in ParseFrontendAttributes()
4215 FrontendAttributes frontend_attributes; in ParseAttributeHelper() local
4216 if (!ParseFrontendAttributes(&frontend_attributes)) { in ParseAttributeHelper()
[all …]
Dwhile_loop_simplifier.cc46 new_while_op->add_frontend_attributes(old_while_op->frontend_attributes()); in CopyFrontendAttributes()
815 const auto& attrs = while_op->frontend_attributes().map(); in TryRemoveWhileLoop()
Dhlo_instruction.cc1001 instruction->set_frontend_attributes(proto.frontend_attributes()); in CreateFromProto()
1700 broadcast->set_frontend_attributes(operand->frontend_attributes()); in CreateBroadcastSequence()
1726 reshaped_operand->set_frontend_attributes(operand->frontend_attributes()); in CreateBroadcastSequence()
1734 broadcast->set_frontend_attributes(operand->frontend_attributes()); in CreateBroadcastSequence()
4056 const FrontendAttributes& frontend_attributes) { in FrontendAttributesToString() argument
4058 frontend_attributes.map().begin(), frontend_attributes.map().end()); in FrontendAttributesToString()
Dhlo_computation.cc1001 if (new_instruction->frontend_attributes().map().empty()) { in ReplaceInstructionWithDifferentShape()
1003 old_instruction->frontend_attributes()); in ReplaceInstructionWithDifferentShape()
Dhlo.proto307 xla.FrontendAttributes frontend_attributes = 68; field
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
Dexport.mlir1493 // correctly in HloModule as frontend_attributes.
1497 …hannel_handle<handle = 1, type = 2>, is_host_transfer = true, mhlo.frontend_attributes = {_xla_hos…
1498 …hannel_handle<handle = 2, type = 3>, is_host_transfer = true, mhlo.frontend_attributes = {_xla_hos…
1505 // CHECK-SAME: frontend_attributes={_xla_host_transfer_original_type="f32",_xla_host_transfer_rende…
1507 // CHECK-SAME: frontend_attributes={_xla_host_transfer_original_type="f32",_xla_host_transfer_rende…
1509 // CHECK-SAME: frontend_attributes={_xla_host_transfer_original_type="f32",_xla_host_transfer_rende…
1511 // CHECK-SAME: frontend_attributes={_xla_host_transfer_original_type="f32",_xla_host_transfer_rende…
1515 // Tests ops with empty frontend attributes do not have frontend_attributes
1520 …hannel_handle<handle = 1, type = 2>, is_host_transfer = true, mhlo.frontend_attributes = {}} : (te…
1524 // CHECK-NOT: frontend_attributes
[all …]
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client_test.py2690 frontend_attributes = xla_client._xla.FrontendAttributes()
2691 frontend_attributes["_xla_host_transfer_rendezvous"] = "undef"
2692 frontend_attributes["_xla_host_transfer_original_type"] = "u32"
2693 frontend_attributes["_xla_host_transfer_is_lower_bits"] = "false"
2694 frontend_attributes["_xla_host_transfer_handler_name"] = "undef"
2695 c.set_frontend_attributes(frontend_attributes)
2747 frontend_attributes = xla_client._xla.FrontendAttributes()
2748 frontend_attributes["_xla_host_transfer_rendezvous"] = "undef"
2749 frontend_attributes["_xla_host_transfer_original_type"] = "u32"
2750 frontend_attributes["_xla_host_transfer_is_lower_bits"] = "false"
[all …]
Dxla_compiler.cc779 py::class_<FrontendAttributes> frontend_attributes(m, "FrontendAttributes"); in BuildXlaCompilerSubmodule() local
780 frontend_attributes.def(py::init<>()) in BuildXlaCompilerSubmodule()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf_communication.cc253 auto frontend_attributes = DictionaryAttr::get( in SetFrontendAttributes() local
257 op->setAttr(kFrontendAttributesAttr, frontend_attributes); in SetFrontendAttributes()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h223 void SetFrontendAttributes(const FrontendAttributes& frontend_attributes) { in SetFrontendAttributes() argument
224 frontend_attributes_ = frontend_attributes; in SetFrontendAttributes()
231 const FrontendAttributes& frontend_attributes) { in SwapFrontendAttributes() argument
233 frontend_attributes_ = frontend_attributes; in SwapFrontendAttributes()
238 const FrontendAttributes& frontend_attributes() const { in frontend_attributes() function
Dxla_builder_test.cc1282 ExpectAttributesMatch(inst->frontend_attributes(), *expected_it); in ExpectInstructionsAttributesMatch()
Dxla_builder.cc481 auto* frontend_attributes = instr_proto->mutable_frontend_attributes(); in SetInstructionFrontendAttribute() local
482 (*frontend_attributes->mutable_map())[attribute] = std::move(value); in SetInstructionFrontendAttribute()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.cc479 xla::FrontendAttributes frontend_attributes; in CreateOpFrontendAttributesFromAttribute() local
483 if (!frontend_attributes_dict) return frontend_attributes; in CreateOpFrontendAttributesFromAttribute()
487 frontend_attributes.mutable_map()->insert( in CreateOpFrontendAttributesFromAttribute()
490 return frontend_attributes; in CreateOpFrontendAttributesFromAttribute()