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

Method GetSize

tensorflow/core/kernels/work_queue_ops.cc:122–126  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

120 }
121
122 Status GetSize(Tensor* size) {
123 std::unique_lock<std::mutex> lock(mu_);
124 size->scalar<int64>().setConstant(static_cast<int64>(queue_.size()));
125 return Status::OK();
126 }
127
128 Status Restore(const Tensor& restorable) {
129 const int64 num_works = restorable.shape().dim_size(0);

Callers 10

mainFunction · 0.45
TranslateInternalMethod · 0.45
mainFunction · 0.45
QuantizeModelFunction · 0.45
mainFunction · 0.45
mainFunction · 0.45
SparsifyModelFunction · 0.45
ComputeMethod · 0.45
ComputeWithTensorBufMethod · 0.45

Calls 1

sizeMethod · 0.45

Tested by 3

mainFunction · 0.36
mainFunction · 0.36
mainFunction · 0.36