Searched defs:tensor_ids (Results 1 – 8 of 8) sorted by relevance
36 std::vector<DependencyGraph::TensorId> tensor_ids{}; in get_tensor_ids() local
38 std::vector<DependencyGraph::TensorId> tensor_ids{}; in get_tensor_ids() local
333 tensor_ids): argument
386 gtl::ArraySlice<int64_t> tensor_ids, in ShouldRecord()901 gtl::ArraySlice<int64_t> tensor_ids, in ShouldRecord()
251 std::vector<int64_t> tensor_ids(tensors.size()); in ShouldRecord() local
244 void PrintTotalBytesOfTensors(const Subgraph& subgraph, const T& tensor_ids, in PrintTotalBytesOfTensors()
1977 std::vector<int64_t> tensor_ids; local1998 bool TensorShapesAndDtypes(PyObject* tensors, std::vector<int64_t>* tensor_ids,2040 std::vector<int64_t> tensor_ids; local2075 std::vector<int64_t> tensor_ids; local
1550 std::vector<TensorId> tensor_ids; in Print() local