Searched defs:ShapeSizeBytesFunction (Results 1 – 3 of 3) sorted by relevance
67 HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const override { in ShapeSizeBytesFunction() function
184 HloCostAnalysis::ShapeSizeFunction InterpreterCompiler::ShapeSizeBytesFunction() in ShapeSizeBytesFunction() function in xla::interpreter::InterpreterCompiler
946 HloCostAnalysis::ShapeSizeFunction CpuCompiler::ShapeSizeBytesFunction() const { in ShapeSizeBytesFunction() function in xla::cpu::CpuCompiler