Searched refs:record_tensor_accesses (Results 1 – 5 of 5) sorted by relevance
324 params.record_tensor_accesses = false; in TEST_F()325 params.device = new DummyDevice(env, params.record_tensor_accesses); in TEST_F()349 params.record_tensor_accesses = true; in TEST_F()350 params.device = new DummyDevice(env, params.record_tensor_accesses); in TEST_F()377 params.record_tensor_accesses = false; in TEST_F()378 params.device = new DummyDevice(env, params.record_tensor_accesses); in TEST_F()658 params.record_tensor_accesses = false; in TEST_F()660 new DummyDevice(env, params.record_tensor_accesses)); in TEST_F()
617 bool record_tensor_accesses = false; member1535 params_->record_tensor_accesses); in record_tensor_reference()1536 if (params_->record_tensor_accesses) { in record_tensor_reference()1543 if (params_->record_tensor_accesses) { in retrieve_accessed_tensors()
296 if (params_->record_tensor_accesses) { in OpKernelContext()307 if (params_->record_tensor_accesses) referenced_tensors_.Destroy(); in ~OpKernelContext()
208 params.record_tensor_accesses = false; // TODO(mrry): Too severe? in RunAsync()
1627 params.record_tensor_accesses = impl_->device_record_tensor_accesses_; in Process()