Searched refs:ShapeSizeFunction (Results 1 – 19 of 19) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | while_loop_expensive_invariant_code_motion.h | 32 using ShapeSizeFunction = std::function<int64_t(const Shape&)>;
35 ShapeSizeFunction shape_size_function = ShapeUtil::ByteSizeOfElements)
52 ShapeSizeFunction shape_size_function_;
|
| D | while_loop_invariant_code_motion.h | 31 using ShapeSizeFunction = std::function<int64_t(const Shape&)>;
58 ShapeSizeFunction shape_size_function = ShapeUtil::ByteSizeOfElements)
83 ShapeSizeFunction shape_size_function_;
|
| D | hlo_rematerialization.h | 42 using ShapeSizeFunction = std::function<int64_t(const Shape&)>;
85 const ShapeSizeFunction& size_function, int64_t memory_limit_bytes,
167 const ShapeSizeFunction size_function_;
|
| D | hlo_cost_analysis.h | 47 using ShapeSizeFunction = std::function<int64_t(const Shape&)>; variable
62 ShapeSizeFunction shape_size;
86 explicit HloCostAnalysis(ShapeSizeFunction shape_size,
|
| D | compiler.h | 357 virtual HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const = 0;
362 HloCostAnalysis::ShapeSizeFunction shape_size = ShapeSizeBytesFunction(); in BufferSizeBytesFunction()
|
| D | hlo_rematerialization.cc | 499 const HloRematerialization::ShapeSizeFunction& size_function,
771 const HloRematerialization::ShapeSizeFunction& size_function_;
794 const HloRematerialization::ShapeSizeFunction& size_function, in MemoryUsageTracker()
|
| D | hlo_cost_analysis.cc | 43 HloCostAnalysis::HloCostAnalysis(ShapeSizeFunction shape_size, in HloCostAnalysis()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | parallel_task_assignment.h | 43 const HloCostAnalysis::ShapeSizeFunction& shape_size,
70 const HloCostAnalysis::ShapeSizeFunction& shape_size, in ParallelTaskAssigner()
108 HloCostAnalysis::ShapeSizeFunction shape_size_function_;
|
| D | parallel_task_assignment.cc | 35 const HloCostAnalysis::ShapeSizeFunction& shape_size) in SimpleCostModel()
51 const HloCostAnalysis::ShapeSizeFunction shape_size_;
57 const HloCostAnalysis::ShapeSizeFunction& shape_size, in DefaultCostModel()
108 const HloCostAnalysis::ShapeSizeFunction shape_size_;
114 const HloCostAnalysis::ShapeSizeFunction& shape_size, HloModule* module, in ParallelTaskAssignment()
|
| D | cpu_xfeed.h | 42 HloCostAnalysis::ShapeSizeFunction shape_size_fn);
|
| D | parallel_task_assignment_test.cc | 29 const HloCostAnalysis::ShapeSizeFunction shape_size_func_ =
|
| D | cpu_compiler.h | 155 HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const override;
|
| D | cpu_xfeed.cc | 278 HloCostAnalysis::ShapeSizeFunction shape_size_fn) { in ReadDynamicShapesOnCpu()
|
| D | cpu_compiler.cc | 1580 HloCostAnalysis::ShapeSizeFunction CpuCompiler::ShapeSizeBytesFunction() const { in ShapeSizeBytesFunction()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | compiler.h | 60 HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const override;
|
| D | compiler.cc | 170 HloCostAnalysis::ShapeSizeFunction InterpreterCompiler::ShapeSizeBytesFunction() in ShapeSizeBytesFunction()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_compiler.h | 111 HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const override;
|
| D | gpu_compiler.cc | 1512 HloCostAnalysis::ShapeSizeFunction GpuCompiler::ShapeSizeBytesFunction() const { in ShapeSizeBytesFunction()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_on_demand_compiler.cc | 194 HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const override { in ShapeSizeBytesFunction()
|