Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_compiler.h107 HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const override { in ShapeSizeBytesFunction() function
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dcompiler.cc166 HloCostAnalysis::ShapeSizeFunction InterpreterCompiler::ShapeSizeBytesFunction() in ShapeSizeBytesFunction() function in xla::interpreter::InterpreterCompiler
/external/tensorflow/tensorflow/core/tpu/
Dtpu_on_demand_compiler.cc193 HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const override { in ShapeSizeBytesFunction() function in xla::__anonbad6196c0111::TpuCompiler
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_compiler.cc1066 HloCostAnalysis::ShapeSizeFunction CpuCompiler::ShapeSizeBytesFunction() const { in ShapeSizeBytesFunction() function in xla::cpu::CpuCompiler