Home
last modified time | relevance | path

Searched refs:ShapeSizeFunction (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_invariant_code_motion.h30 using ShapeSizeFunction = std::function<int64(const Shape&)>;
44 ShapeSizeFunction shape_size_function = ShapeUtil::ByteSizeOfElements)
62 ShapeSizeFunction shape_size_function_;
Dwhile_loop_expensive_invariant_code_motion.h32 using ShapeSizeFunction = std::function<int64(const Shape&)>;
35 ShapeSizeFunction shape_size_function = ShapeUtil::ByteSizeOfElements)
49 ShapeSizeFunction shape_size_function_;
Dhlo_rematerialization.h41 using ShapeSizeFunction = std::function<int64(const Shape&)>;
84 const ShapeSizeFunction& size_function, int64 memory_limit_bytes,
138 const ShapeSizeFunction size_function_;
Dhlo_cost_analysis.h50 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_;
Dcompiler.h291 virtual HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const = 0;
296 HloCostAnalysis::ShapeSizeFunction shape_size = ShapeSizeBytesFunction(); in BufferSizeBytesFunction()
Dhlo_cost_analysis.cc40 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()
Dhlo_rematerialization.cc495 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/
Dparallel_task_assignment.h42 const HloCostAnalysis::ShapeSizeFunction& shape_size,
69 const HloCostAnalysis::ShapeSizeFunction& shape_size, in ParallelTaskAssigner()
102 HloCostAnalysis::ShapeSizeFunction shape_size_function_;
Dparallel_task_assignment.cc34 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()
Dparallel_task_assignment_test.cc28 const HloCostAnalysis::ShapeSizeFunction shape_size_func_ =
Dcpu_compiler.h153 HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const override;
Dcpu_compiler.cc1032 HloCostAnalysis::ShapeSizeFunction CpuCompiler::ShapeSizeBytesFunction() const { in ShapeSizeBytesFunction()
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dcompiler.h61 HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const override;
Dcompiler.cc165 HloCostAnalysis::ShapeSizeFunction InterpreterCompiler::ShapeSizeBytesFunction() in ShapeSizeBytesFunction()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_compiler.h112 HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const override { in ShapeSizeBytesFunction()
/external/tensorflow/tensorflow/core/tpu/
Dtpu_on_demand_compiler.cc358 HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const override { in ShapeSizeBytesFunction()