Home
last modified time | relevance | path

Searched refs:IsSameAs (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dtracked_device_buffer_test.cc77 EXPECT_TRUE(expected_it->IsSameAs(it->second)); in TEST()
83 EXPECT_TRUE(expected_it->IsSameAs(it->second)); in TEST()
89 EXPECT_TRUE(expected_it->IsSameAs(it->second)); in TEST()
/external/tensorflow/tensorflow/compiler/xla/service/
Dshaped_buffer_test.cc142 EXPECT_TRUE(buffers.find(orig_index)->second.IsSameAs(buffer)); in TEST()
149 EXPECT_TRUE(buffers.find(index)->second.IsSameAs(buffer)); in TEST()
/external/tensorflow/tensorflow/stream_executor/
Ddevice_memory.h95 bool IsSameAs(const DeviceMemoryBase &other) const { in IsSameAs() function
/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_executable_interface.cc206 return index_addr_pair.second.IsSameAs( in ExecuteAsyncOnStream()
/external/tensorflow/tensorflow/compiler/xrt/
Dxrt_state.cc641 if (dest_buffer->allocation().IsSameAs(source_buffer->allocation()) && in AliasBufferFrom()