MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / EnqueueConst

Function EnqueueConst

tensorflow/compiler/xrt/client/xrt_tf_client.cc:524–534  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

522}
523
524XrtTensorHandle EnqueueConst(XrtTfContext* context, int device_id,
525 TensorProto value, bool host_memory) {
526 protobuf::Map<std::string, AttrValue> attrs;
527 attrs["value"] = MakeAttrValue(value);
528 attrs["dtype"] = MakeAttrValue(value.dtype());
529
530 return std::move(context->EnqueueOp(host_memory ? "HostConst" : "Const",
531 /*inputs=*/{},
532 /*output_arity=*/1, std::move(attrs),
533 device_id)[0]);
534}
535
536} // namespace tensorflow

Callers 3

xrt_client.ccFile · 0.85
ExecuteMethod · 0.85
InitializeTPUMethod · 0.85

Calls 3

MakeAttrValueFunction · 0.85
dtypeMethod · 0.45
EnqueueOpMethod · 0.45

Tested by

no test coverage detected