Home
last modified time | relevance | path

Searched defs:ShapeSizeBytesFunction (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dcompiler.cc170 HloCostAnalysis::ShapeSizeFunction InterpreterCompiler::ShapeSizeBytesFunction() in ShapeSizeBytesFunction() function in xla::interpreter::InterpreterCompiler
/external/tensorflow/tensorflow/core/tpu/
Dtpu_on_demand_compiler.cc194 HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const override { in ShapeSizeBytesFunction() function in xla::__anon3e43d5360111::TpuCompiler
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_compiler.cc1580 HloCostAnalysis::ShapeSizeFunction CpuCompiler::ShapeSizeBytesFunction() const { in ShapeSizeBytesFunction() function in xla::cpu::CpuCompiler
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_compiler.cc1512 HloCostAnalysis::ShapeSizeFunction GpuCompiler::ShapeSizeBytesFunction() const { in ShapeSizeBytesFunction() function in xla::gpu::GpuCompiler