Searched refs:TFProfTensor (Results 1 – 7 of 7) sorted by relevance
20 void TFProfTensor::Display(string* formatted_str, in Display()35 void TFProfTensor::Build() { in Build()
36 class TFProfTensor {38 explicit TFProfTensor(std::unique_ptr<Tensor> tensor) in TFProfTensor() function
56 std::unique_ptr<TFProfTensor>* tensor);
54 std::unique_ptr<TFProfTensor>* tensor) { in LookUpCheckPoint()66 tensor->reset(new TFProfTensor(std::move(out_tensor))); in LookUpCheckPoint()
55 std::unique_ptr<TFProfTensor>* tensor);
179 std::unique_ptr<TFProfTensor> tfprof_tensor; in PrintScope()
196 std::unique_ptr<TFProfTensor> tfprof_tensor; in PrintGraph()