Searched refs:ShapeSizeBytesFunction (Results 1 – 15 of 15) sorted by relevance
61 HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const override;
165 HloCostAnalysis::ShapeSizeFunction InterpreterCompiler::ShapeSizeBytesFunction() in ShapeSizeBytesFunction() function in xla::interpreter::InterpreterCompiler
291 virtual HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const = 0;296 HloCostAnalysis::ShapeSizeFunction shape_size = ShapeSizeBytesFunction(); in BufferSizeBytesFunction()
225 auto shape_size_fn = compiler->ShapeSizeBytesFunction(); in ReadDynamicShapes()
1120 execute_backend_->compiler()->ShapeSizeBytesFunction()); in GetComputationGraphStats()
153 HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const override;
441 max_parallelism, ShapeSizeBytesFunction(), target_machine_features); in RunHloPassesAfterLayoutAssn()1032 HloCostAnalysis::ShapeSizeFunction CpuCompiler::ShapeSizeBytesFunction() const { in ShapeSizeBytesFunction() function in xla::cpu::CpuCompiler
112 HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const override { in ShapeSizeBytesFunction() function
884 HloCostAnalysis cost_analysis(ShapeSizeBytesFunction()); in RunBackend()
170 backend_->compiler()->ShapeSizeBytesFunction()); in CreateNewVerifiedModule()
226 local_client_->backend().compiler()->ShapeSizeBytesFunction()); in ParseAndReturnVerifiedModule()
124 backend().compiler()->ShapeSizeBytesFunction()); in CreateNewVerifiedModule()140 backend().compiler()->ShapeSizeBytesFunction()); in ParseAndReturnVerifiedModule()
163 auto shape_size_fn = compiler->ShapeSizeBytesFunction(); in UpdateMetadata()208 auto shape_size_fn = compiler->ShapeSizeBytesFunction(); in UpdateDynamicInputs()
358 HloCostAnalysis::ShapeSizeFunction ShapeSizeBytesFunction() const override { in ShapeSizeBytesFunction() function in xla::__anoncef202870111::TpuCompiler
263 client_->backend().compiler()->ShapeSizeBytesFunction()); in GetHloCostAnalysis()