Home
last modified time | relevance | path

Searched refs:ShapeSize (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_cost_analysis_test.cc43 int64 ShapeSize(const Shape& shape) { in ShapeSize() function
147 HloCostAnalysis analysis(ShapeSize); in TEST_F()
176 HloCostAnalysis analysis(ShapeSize); in TEST_F()
205 HloCostAnalysis analysis(ShapeSize); in TEST_F()
228 HloCostAnalysis analysis(ShapeSize); in TEST_F()
249 HloCostAnalysis analysis(ShapeSize); in TEST_F()
275 HloCostAnalysis analysis(ShapeSize); in TEST_F()
304 HloCostAnalysis analysis(ShapeSize); in TEST_F()
325 HloCostAnalysis analysis(ShapeSize); in TEST_F()
343 HloCostAnalysis analysis(ShapeSize); in TEST_F()
[all …]
Dhlo_verifier.h178 int64 ShapeSize(const Shape& shape) const { in ShapeSize() function
Dhlo_verifier.cc1499 return target_metadata_->ShapeSize(shape); in Run()