Searched refs:ShapeSize (Results 1 – 9 of 9) sorted by relevance
42 int64 ShapeSize(const Shape& shape) { in ShapeSize() function146 HloCostAnalysis analysis(ShapeSize); in TEST_F()180 HloCostAnalysis analysis(ShapeSize); in TEST_F()216 HloCostAnalysis analysis(ShapeSize); in TEST_F()246 HloCostAnalysis analysis(ShapeSize); in TEST_F()273 HloCostAnalysis analysis(ShapeSize); in TEST_F()303 HloCostAnalysis analysis(ShapeSize); in TEST_F()337 HloCostAnalysis analysis(ShapeSize); in TEST_F()360 HloCostAnalysis analysis(ShapeSize); in TEST_F()383 HloCostAnalysis analysis(ShapeSize); in TEST_F()[all …]
215 int64 ShapeSize(const Shape& shape) const { in ShapeSize() function
34 int64 ShapeSize(const Shape& shape) { in ShapeSize() function48 HloCostAnalysis hlo_cost_analysis(ShapeSize); in AssignMemorySpaceUsingCostAnalysis()4110 HloCostAnalysis hlo_cost_analysis(ShapeSize); in TEST_P()4116 return ShapeSize(shape) / 16; in TEST_P()5258 HloCostAnalysis hlo_cost_analysis(ShapeSize); in TEST_F()5355 HloCostAnalysis hlo_cost_analysis(ShapeSize); in TEST_F()5435 HloCostAnalysis hlo_cost_analysis(ShapeSize); in TEST_F()5501 HloCostAnalysis hlo_cost_analysis(ShapeSize); in TEST_F()
1961 return target_metadata_->ShapeSize(shape); in Run()
61 int64 ShapeSize(const Shape& shape) override;
84 virtual int64 ShapeSize(const Shape& shape) = 0;
118 int64 TpuExecutable::ShapeSize(const Shape& shape) { in ShapeSize() function in xla::TpuExecutable
101 int64 allocation_bytes = ShapeSize(ShapeUtil::GetSubshape( in AllocateOutputMemoryWithInputReuse()
202 int64 ShapeSize(const Shape& shape) override { in ShapeSize() function in xla::__anoncef202870111::TpuExecutable