Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/python/
Dxrt.cc48 const string& session_target) in XrtAllocation() argument
49 : handle_(handle), shape_(shape), session_target_(session_target) {} in XrtAllocation()
75 const Literal& argument, const string& session_target) { in FromLiteral() argument
85 tensorflow::ClientSession session(root, session_target); in FromLiteral()
92 return new XrtAllocation(handle, argument.shape(), session_target); in FromLiteral()
146 const Shape& result_shape, const string& session_target) { in CompileForXrt() argument
167 tensorflow::ClientSession session(root, session_target); in CompileForXrt()
174 return new XrtExecutable(program_shape, handle, session_target); in CompileForXrt()
178 const string& session_target) in XrtExecutable() argument
181 session_target_(session_target) {} in XrtExecutable()
[all …]
Dxrt.h38 const string& session_target);
40 XrtAllocation(int64 handle, Shape shape, const string& session_target);
79 XrtAllocation* allocation, const string& session_target);
89 const Shape& result_shape, const string& session_target);
94 const string& session_target);