Searched refs:ShapeSizeFunction (Results 1 – 16 of 16) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | while_loop_invariant_code_motion.h | 30 using ShapeSizeFunction = std::function<int64(const Shape&)>; 44 ShapeSizeFunction shape_size_function = ShapeUtil::ByteSizeOfElements) 62 ShapeSizeFunction shape_size_function_;
|
D | while_loop_expensive_invariant_code_motion.h | 32 using ShapeSizeFunction = std::function<int64(const Shape&)>; 35 ShapeSizeFunction shape_size_function = ShapeUtil::ByteSizeOfElements) 49 ShapeSizeFunction shape_size_function_;
|
D | hlo_rematerialization.h | 41 using ShapeSizeFunction = std::function<int64(const Shape&)>; 84 const ShapeSizeFunction& size_function, int64 memory_limit_bytes, 138 const ShapeSizeFunction size_function_;
|
D | hlo_cost_analysis.h | 50 using ShapeSizeFunction = std::function<int64(const Shape&)>; variable 51 explicit HloCostAnalysis(const ShapeSizeFunction& shape_size); 198 HloCostAnalysis(const ShapeSizeFunction& shape_size, 202 const ShapeSizeFunction& shape_size, const Properties& per_second_rates); 241 const ShapeSizeFunction shape_size_;
|
D | compiler.h | 291 virtual HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const = 0; 296 HloCostAnalysis::ShapeSizeFunction shape_size = ShapeSizeBytesFunction(); in BufferSizeBytesFunction()
|
D | hlo_cost_analysis.cc | 40 HloCostAnalysis::HloCostAnalysis(const ShapeSizeFunction& shape_size) in HloCostAnalysis() 43 HloCostAnalysis::HloCostAnalysis(const ShapeSizeFunction& shape_size, in HloCostAnalysis() 1147 const ShapeSizeFunction& shape_size, const Properties& per_second_rates) { in CreateNestedCostAnalysis()
|
D | hlo_rematerialization.cc | 495 const HloRematerialization::ShapeSizeFunction& size_function, 760 const HloRematerialization::ShapeSizeFunction& size_function_; 783 const HloRematerialization::ShapeSizeFunction& size_function, in MemoryUsageTracker()
|
/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 | 34 const HloCostAnalysis::ShapeSizeFunction& shape_size) in SimpleCostModel() 49 const HloCostAnalysis::ShapeSizeFunction shape_size_; 55 const HloCostAnalysis::ShapeSizeFunction& shape_size, in DefaultCostModel() 105 const HloCostAnalysis::ShapeSizeFunction shape_size_; 111 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 | 153 HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const override;
|
D | cpu_compiler.cc | 1032 HloCostAnalysis::ShapeSizeFunction CpuCompiler::ShapeSizeBytesFunction() const { in ShapeSizeBytesFunction()
|
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
D | compiler.h | 61 HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const override;
|
D | compiler.cc | 165 HloCostAnalysis::ShapeSizeFunction InterpreterCompiler::ShapeSizeBytesFunction() in ShapeSizeBytesFunction()
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | gpu_compiler.h | 112 HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const override { in ShapeSizeBytesFunction()
|
/external/tensorflow/tensorflow/core/tpu/ |
D | tpu_on_demand_compiler.cc | 358 HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const override { in ShapeSizeBytesFunction()
|