Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Dpy_executable.h92 int64_t SizeOfGeneratedCodeInBytes() const { in SizeOfGeneratedCodeInBytes() function
93 return executable_->SizeOfGeneratedCodeInBytes(); in SizeOfGeneratedCodeInBytes()
H A Dxla.cc375 &PyExecutable::SizeOfGeneratedCodeInBytes) in PYBIND11_MODULE()
H A Dpy_client.cc436 executable->SizeOfGeneratedCodeInBytes(), nullptr}; in HeapProfile()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dexecutable.cc238 executable->SizeOfGeneratedCodeInBytes(); in ExecuteWrapperAfterExecution()
300 int64_t Executable::SizeOfGeneratedCodeInBytes() const { return -1; } in SizeOfGeneratedCodeInBytes() function in xla::Executable
H A Dexecutable.h367 virtual int64_t SizeOfGeneratedCodeInBytes() const;
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_executable.cc404 int64_t CpuExecutable::SizeOfGeneratedCodeInBytes() const { in SizeOfGeneratedCodeInBytes() function in xla::cpu::CpuExecutable
405 return jit_->SizeOfGeneratedCodeInBytes(); in SizeOfGeneratedCodeInBytes()
H A Dcpu_executable.h89 int64_t SizeOfGeneratedCodeInBytes() const override;
H A Dsimple_orc_jit.h98 int64_t SizeOfGeneratedCodeInBytes() const { in SizeOfGeneratedCodeInBytes() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dpjrt_executable.h55 virtual int64_t SizeOfGeneratedCodeInBytes() const = 0;
H A Dpjrt_c_api_client.h436 int64_t SizeOfGeneratedCodeInBytes() const override { in SizeOfGeneratedCodeInBytes() function
438 return wrapped()->SizeOfGeneratedCodeInBytes(); in SizeOfGeneratedCodeInBytes()
H A Dtfrt_cpu_pjrt_client.h492 int64_t SizeOfGeneratedCodeInBytes() const override { in SizeOfGeneratedCodeInBytes() function
493 return cpu_executable_->SizeOfGeneratedCodeInBytes(); in SizeOfGeneratedCodeInBytes()
H A Dpjrt_stream_executor_client.h734 int64_t SizeOfGeneratedCodeInBytes() const override { in SizeOfGeneratedCodeInBytes() function
737 size += executable->executable()->SizeOfGeneratedCodeInBytes(); in SizeOfGeneratedCodeInBytes()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_executable.h163 int64_t SizeOfGeneratedCodeInBytes() const override;
H A Dgpu_executable.cc870 int64_t GpuExecutable::SizeOfGeneratedCodeInBytes() const { in SizeOfGeneratedCodeInBytes() function in xla::gpu::GpuExecutable
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
H A Dtpu_client_extension.cc239 &PyTpuExecutable::SizeOfGeneratedCodeInBytes) in PYBIND11_MODULE()
H A Dtpu_client.h366 int64_t SizeOfGeneratedCodeInBytes() const { in SizeOfGeneratedCodeInBytes() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dlocal_client_execute_test.cc795 DISABLED_ON_INTERPRETER(SizeOfGeneratedCodeInBytes)) { in XLA_TEST_F() argument
813 EXPECT_GT(executables.front()->executable()->SizeOfGeneratedCodeInBytes(), in XLA_TEST_F()