Searched refs:device_contexts (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device.cc | 386 TF_ASSIGN_OR_RETURN(auto device_contexts, GetDeviceContextLocked()); in TryGetDeviceContext()
387 device_contexts.first->Ref(); in TryGetDeviceContext()
388 *out_context = device_contexts.first; in TryGetDeviceContext()
503 std::pair<XlaDeviceContext*, XlaDeviceContext*> device_contexts; in MakeTensorFromProto() local
506 TF_ASSIGN_OR_RETURN(device_contexts, GetDeviceContextLocked()); in MakeTensorFromProto()
508 return MakeTensorFromProto(device_contexts.first, tensor_proto, alloc_attrs, in MakeTensorFromProto()
516 std::pair<XlaDeviceContext*, XlaDeviceContext*> device_contexts; in MakeFastMemTensorFromProto() local
519 TF_ASSIGN_OR_RETURN(device_contexts, GetDeviceContextLocked()); in MakeFastMemTensorFromProto()
521 return MakeTensorFromProto(device_contexts.second, tensor_proto, alloc_attrs, in MakeFastMemTensorFromProto()
|