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

Method op_device_context

tensorflow/core/framework/op_kernel.h:1731–1735  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1729
1730template <typename T>
1731T* OpKernelContext::op_device_context() {
1732 static_assert(std::is_base_of<DeviceContext, T>::value,
1733 "T is not a subclass of DeviceContext");
1734 return static_cast<T*>(op_device_context());
1735}
1736
1737inline const Tensor& OpInputList::operator[](int i) const {
1738 DCHECK_GE(i, 0);

Callers 15

ComputeAsyncMethod · 0.45
ComputeAsyncMethod · 0.45
ComputeAsyncMethod · 0.45
ComputeAsyncMethod · 0.45
operator()Method · 0.45
launchMethod · 0.45
DoWorkMethod · 0.45
PopulateInputsMethod · 0.45
PopulateOutputsMethod · 0.45
RunMethod · 0.45
CompileMethod · 0.45
GetAllocatorFunction · 0.45

Calls

no outgoing calls

Tested by 2

ComputeMethod · 0.36
ComputeMethod · 0.36