Home
last modified time | relevance | path

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

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