HomeSort by relevance Sort by last modified time
    Searched refs:CpuExecutable (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_executable.h 48 class CpuExecutable : public Executable {
50 CpuExecutable(std::unique_ptr<SimpleOrcJIT> jit,
56 ~CpuExecutable() override {}
134 // CpuExecutable. We save a string instead of an llvm::Module* because leaving
144 TF_DISALLOW_COPY_AND_ASSIGN(CpuExecutable);
cpu_executable.cc 53 CpuExecutable::CpuExecutable(
76 Status CpuExecutable::AllocateBuffers(
124 Status CpuExecutable::ExecuteComputeFunction(
238 VLOG(3) << "CpuExecutable deallocating buffer #" << i << " ["
248 StatusOr<std::unique_ptr<ShapedBuffer>> CpuExecutable::CreateResultShapedBuffer(
287 StatusOr<std::unique_ptr<ShapedBuffer>> CpuExecutable::ExecuteOnStream(
316 StatusOr<std::unique_ptr<ShapedBuffer>> CpuExecutable::ExecuteAsyncOnStream(
355 /*static*/ int64 CpuExecutable::ShapeSizeBytes(const Shape& shape) {
363 const PointsToSet& CpuExecutable::GetRootPointsToSet() const
    [all...]
cpu_compiler.cc 563 int64 size = CpuExecutable::ShapeSizeBytes(instruction->shape());
627 static_cast<CpuExecutable&>(*cpu_executable)
701 cpu_executable.reset(new CpuExecutable(
706 static_cast<CpuExecutable&>(*cpu_executable)
    [all...]
  /external/tensorflow/tensorflow/compiler/tf2xla/
xla_jit_compiled_cpu_function.cc 143 // Compile the executable. The static_cast to the CpuExecutable subclass is
148 const xla::cpu::CpuExecutable* cpu_executable =
149 static_cast<xla::cpu::CpuExecutable*>(executable->executable());

Completed in 189 milliseconds