Searched refs:ShapeSizeFunction (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | optimize_input_output_buffer_alias.h | 48 using ShapeSizeFunction = std::function<int64(const Shape&)>; variable
51 OptimizeInputOutputBufferAlias(ShapeSizeFunction size_func) in OptimizeInputOutputBufferAlias()
66 ShapeSizeFunction size_func_ = nullptr;
|
| D | hlo_rematerialization.h | 40 using ShapeSizeFunction = std::function<int64(const Shape&)>;
60 HloRematerialization(const ShapeSizeFunction& size_function, in HloRematerialization()
102 const ShapeSizeFunction size_function_;
|
| D | hlo_cost_analysis.h | 49 using ShapeSizeFunction = std::function<int64(const Shape&)>; variable
50 explicit HloCostAnalysis(const ShapeSizeFunction& shape_size);
156 HloCostAnalysis(const ShapeSizeFunction& shape_size,
202 const ShapeSizeFunction shape_size_;
|
| D | compiler.h | 246 virtual HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const = 0;
251 HloCostAnalysis::ShapeSizeFunction shape_size = ShapeSizeBytesFunction(); in BufferSizeBytesFunction()
|
| D | hlo_rematerialization.cc | 329 const HloRematerialization::ShapeSizeFunction& size_function,
416 const HloRematerialization::ShapeSizeFunction& size_function, in CreateBufferFromLogicalBuffer()
510 const HloRematerialization::ShapeSizeFunction& size_function, in MemoryUsageTracker()
|
| D | hlo_cost_analysis.cc | 35 HloCostAnalysis::HloCostAnalysis(const ShapeSizeFunction& shape_size) in HloCostAnalysis()
38 HloCostAnalysis::HloCostAnalysis(const ShapeSizeFunction& shape_size, in HloCostAnalysis()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | parallel_task_assignment.h | 42 const HloCostAnalysis::ShapeSizeFunction& shape_size,
69 const HloCostAnalysis::ShapeSizeFunction& shape_size, in ParallelTaskAssigner()
102 HloCostAnalysis::ShapeSizeFunction shape_size_function_;
|
| D | parallel_task_assignment.cc | 33 const HloCostAnalysis::ShapeSizeFunction& shape_size) in SimpleCostModel()
48 const HloCostAnalysis::ShapeSizeFunction shape_size_;
54 const HloCostAnalysis::ShapeSizeFunction& shape_size, in DefaultCostModel()
103 const HloCostAnalysis::ShapeSizeFunction shape_size_;
109 const HloCostAnalysis::ShapeSizeFunction& shape_size, HloModule* module, in ParallelTaskAssignment()
|
| D | parallel_task_assignment_test.cc | 28 const HloCostAnalysis::ShapeSizeFunction shape_size_func_ =
|
| D | cpu_compiler.h | 150 HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const override;
|
| D | cpu_compiler.cc | 946 HloCostAnalysis::ShapeSizeFunction CpuCompiler::ShapeSizeBytesFunction() const { in ShapeSizeBytesFunction()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | compiler.h | 71 HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const override;
|
| D | compiler.cc | 184 HloCostAnalysis::ShapeSizeFunction InterpreterCompiler::ShapeSizeBytesFunction() in ShapeSizeBytesFunction()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | nvptx_compiler.h | 67 HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const override { in ShapeSizeBytesFunction()
|