Home
last modified time | relevance | path

Searched refs:SizeOfGeneratedCodeInBytes (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/python/
Dpy_executable.h53 int64 SizeOfGeneratedCodeInBytes() const { in SizeOfGeneratedCodeInBytes() function
54 return executable_->SizeOfGeneratedCodeInBytes(); in SizeOfGeneratedCodeInBytes()
Dpy_client.cc283 executable->SizeOfGeneratedCodeInBytes(), nullptr}; in HeapProfile()
Dxla.cc354 &PyExecutable::SizeOfGeneratedCodeInBytes) in PYBIND11_MODULE()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_executable.h84 int64 SizeOfGeneratedCodeInBytes() const override;
Dsimple_orc_jit.h97 int64 SizeOfGeneratedCodeInBytes() const { in SizeOfGeneratedCodeInBytes() function
Dcpu_executable.cc413 int64 CpuExecutable::SizeOfGeneratedCodeInBytes() const { in SizeOfGeneratedCodeInBytes() function in xla::cpu::CpuExecutable
414 return jit_->SizeOfGeneratedCodeInBytes(); in SizeOfGeneratedCodeInBytes()
/external/tensorflow/tensorflow/compiler/xla/service/
Dexecutable.cc239 executable->SizeOfGeneratedCodeInBytes(); in ExecuteWrapperAfterExecution()
301 int64 Executable::SizeOfGeneratedCodeInBytes() const { return -1; } in SizeOfGeneratedCodeInBytes() function in xla::Executable
Dexecutable.h369 virtual int64 SizeOfGeneratedCodeInBytes() const;
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_executable.h93 int64 SizeOfGeneratedCodeInBytes() const override;
Dgpu_executable.cc567 int64 GpuExecutable::SizeOfGeneratedCodeInBytes() const { in SizeOfGeneratedCodeInBytes() function in xla::gpu::GpuExecutable
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_stream_executor_client.h625 int64 SizeOfGeneratedCodeInBytes() const override { in SizeOfGeneratedCodeInBytes() function
628 size += executable->executable()->SizeOfGeneratedCodeInBytes(); in SizeOfGeneratedCodeInBytes()
Dpjrt_client.h432 virtual int64 SizeOfGeneratedCodeInBytes() const = 0;
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client_extension.cc189 &PyTpuExecutable::SizeOfGeneratedCodeInBytes) in PYBIND11_MODULE()
Dtpu_client.h317 int64 SizeOfGeneratedCodeInBytes() const { in SizeOfGeneratedCodeInBytes() function
/external/tensorflow/tensorflow/compiler/xla/tests/
Dlocal_client_execute_test.cc810 DISABLED_ON_INTERPRETER(SizeOfGeneratedCodeInBytes)) { in XLA_TEST_F() argument
828 EXPECT_GT(executables.front()->executable()->SizeOfGeneratedCodeInBytes(), in XLA_TEST_F()