Searched refs:retrieve_accessed_tensors (Results 1 – 3 of 3) sorted by relevance
339 ctx->retrieve_accessed_tensors(&referenced_tensors); in TEST_F()364 ctx->retrieve_accessed_tensors(&referenced_tensors); in TEST_F()
1160 void retrieve_accessed_tensors(TensorReferenceVector* out_vector);1541 inline void OpKernelContext::retrieve_accessed_tensors( in retrieve_accessed_tensors() function
1779 state->ctx.retrieve_accessed_tensors(&accessed); in Process()1835 ctx.retrieve_accessed_tensors(&accessed_tensors); in Process()