Searched refs:CpuExecutable (Results 1 – 7 of 7) sorted by relevance
53 CpuExecutable::CpuExecutable( in CpuExecutable() function in xla::cpu::CpuExecutable88 CpuExecutable::~CpuExecutable() { in ~CpuExecutable()130 StatusOr<std::vector<MaybeOwningDeviceMemory>> CpuExecutable::CreateBufferTable( in CreateBufferTable()153 Status CpuExecutable::ExecuteComputeFunction( in ExecuteComputeFunction()218 StatusOr<ExecutionOutput> CpuExecutable::CreateResultShapedBuffer( in CreateResultShapedBuffer()318 StatusOr<ExecutionOutput> CpuExecutable::ExecuteAsyncOnStream( in ExecuteAsyncOnStream()366 CpuExecutable* executable; in ExecuteAsyncOnStream()386 /*static*/ int64_t CpuExecutable::ShapeSizeBytes(const Shape& shape) { in ShapeSizeBytes()399 const InstructionValueSet& CpuExecutable::GetRootValueSet() const { in GetRootValueSet()404 int64_t CpuExecutable::SizeOfGeneratedCodeInBytes() const { in SizeOfGeneratedCodeInBytes()
46 class CpuExecutable : public Executable {48 CpuExecutable(std::unique_ptr<SimpleOrcJIT> jit,54 ~CpuExecutable() override;148 CpuExecutable(const CpuExecutable&) = delete;149 CpuExecutable& operator=(const CpuExecutable&) = delete;
34 class CpuExecutable; variable178 StatusOr<std::unique_ptr<CpuExecutable>> CompileLegacyCpuExecutable(
1116 StatusOr<std::unique_ptr<CpuExecutable>>1271 auto cpu_executable = std::make_unique<CpuExecutable>( in CompileLegacyCpuExecutable()1307 std::unique_ptr<CpuExecutable> cpu_executable; in RunBackend()1581 return CpuExecutable::ShapeSizeBytes; in ShapeSizeBytesFunction()
30 cpu::CpuExecutable::ShapeSizeBytes;
124 const xla::cpu::CpuExecutable* cpu_executable = in Compile()125 static_cast<xla::cpu::CpuExecutable*>(executable->executable()); in Compile()
238 return std::make_unique<HloCostAnalysis>(cpu::CpuExecutable::ShapeSizeBytes); in GetHloCostAnalysis()393 tensorflow::down_cast<cpu::CpuExecutable*>(cpu_executable.get()); in Compile()884 &shaped_buffer, &ret_shape, cpu::CpuExecutable::ShapeSizeBytes)); in logical_on_device_shape()1178 std::make_unique<HloCostAnalysis>(cpu::CpuExecutable::ShapeSizeBytes); in TfrtCpuExecutable()1432 tensorflow::down_cast<cpu::CpuExecutable*>(cpu_executable_.get()); in ExecuteHelper()