HomeSort by relevance Sort by last modified time
    Searched refs:ShapeSizeBytesFunction (Results 1 - 8 of 8) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
compiler.h 181 virtual HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const = 0;
186 HloCostAnalysis::ShapeSizeFunction shape_size = ShapeSizeBytesFunction();
service.cc     [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/interpreter/
compiler.h 66 HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const override;
compiler.cc 102 HloCostAnalysis::ShapeSizeFunction InterpreterCompiler::ShapeSizeBytesFunction()
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_compiler.h 135 HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const override;
cpu_compiler.cc 311 ShapeSizeBytesFunction());
319 ShapeSizeBytesFunction());
334 ShapeSizeBytesFunction());
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
gpu_compiler.h 68 HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const override {
84 // The size in bytes of a pointer. Used by ShapeSizeBytesFunction.
gpu_compiler.cc 660 HloCostAnalysis cost_analysis(ShapeSizeBytesFunction());

Completed in 446 milliseconds