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.h112 HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const override { in ShapeSizeBytesFunction() function
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dcompiler.cc165 HloCostAnalysis::ShapeSizeFunction InterpreterCompiler::ShapeSizeBytesFunction() in ShapeSizeBytesFunction() function in xla::interpreter::InterpreterCompiler
/external/tensorflow/tensorflow/core/tpu/
Dtpu_on_demand_compiler.cc358 HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const override { in ShapeSizeBytesFunction() function in xla::__anoncef202870111::TpuCompiler
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_compiler.cc1032 HloCostAnalysis::ShapeSizeFunction CpuCompiler::ShapeSizeBytesFunction() const { in ShapeSizeBytesFunction() function in xla::cpu::CpuCompiler