Method is_cpu

oneflow/core/framework/tensor.cpp:118–118 · view source on GitHub ↗

Source from the content-addressed store, hash-verified

116	}
117
118	bool LocalTensor::is_cpu() const { return CHECK_JUST(device())->type() == "cpu"; }
119	bool LocalTensor::is_cuda() const { return CHECK_JUST(device())->type() == "cuda"; }
120
121	Maybe<Tensor> LocalTensor::detach() const {

ndarray_judgment_and_compatibilityFunction · 0.45

PyTensorObject_is_cpuFunction · 0.45

typeMethod · 0.45

device_typeMethod · 0.45

no test coverage detected