Home
last modified time | relevance | path

Searched refs:hlo_cost_analysis (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
DBUILD47 "//tensorflow/compiler/xla/service:hlo_cost_analysis",
116 "//tensorflow/compiler/xla/service:hlo_cost_analysis",
/external/tensorflow/tensorflow/compiler/xla/pjrt/
DBUILD146 "//tensorflow/compiler/xla/service:hlo_cost_analysis",
200 "//tensorflow/compiler/xla/service:hlo_cost_analysis",
/external/tensorflow/tensorflow/compiler/xla/service/
Dmemory_space_assignment_test.cc48 HloCostAnalysis hlo_cost_analysis(ShapeSize); in AssignMemorySpaceUsingCostAnalysis() local
49 hlo_cost_analysis.set_flops_per_second(kFlopsPerSecond); in AssignMemorySpaceUsingCostAnalysis()
50 hlo_cost_analysis.set_bytes_per_second(kBytesPerSecond); in AssignMemorySpaceUsingCostAnalysis()
51 hlo_cost_analysis.set_transcendentals_per_second(kTranscendentalsPerSecond); in AssignMemorySpaceUsingCostAnalysis()
53 TF_CHECK_OK(computation->Accept(&hlo_cost_analysis)); in AssignMemorySpaceUsingCostAnalysis()
57 hlo_cost_analysis, kAsyncCopyBandwidth, in AssignMemorySpaceUsingCostAnalysis()
4110 HloCostAnalysis hlo_cost_analysis(ShapeSize); in TEST_P() local
4113 hlo_cost_analysis, *module)); in TEST_P()
5258 HloCostAnalysis hlo_cost_analysis(ShapeSize); in TEST_F() local
5261 hlo_cost_analysis, *module)); in TEST_F()
[all …]
DBUILD948 ":hlo_cost_analysis",
3001 name = "hlo_cost_analysis",
3002 srcs = ["hlo_cost_analysis.cc"],
3003 hdrs = ["hlo_cost_analysis.h"],
3026 ":hlo_cost_analysis",
3051 ":hlo_cost_analysis",
3068 ":hlo_cost_analysis",
3701 ":hlo_cost_analysis",
/external/tensorflow/tensorflow/core/tpu/
DBUILD268 "//tensorflow/compiler/xla/service:hlo_cost_analysis",
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
DBUILD1047 "//tensorflow/compiler/xla/service:hlo_cost_analysis",
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
DBUILD1070 "//tensorflow/compiler/xla/service:hlo_cost_analysis",