Home
last modified time | relevance | path

Searched refs:GetComputationShape (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dreplay_test.cc55 client_->GetComputationShape(computation).ConsumeValueOrDie(); in TEST_F()
57 client_->GetComputationShape(replayed).ConsumeValueOrDie(); in TEST_F()
88 client_->GetComputationShape(computation).ConsumeValueOrDie(); in XLA_TEST_F()
90 client_->GetComputationShape(replayed).ConsumeValueOrDie(); in XLA_TEST_F()
135 client_->GetComputationShape(computation).ConsumeValueOrDie(); in TEST_F()
137 client_->GetComputationShape(replayed).ConsumeValueOrDie(); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/tools/
Dshow_signature.cc55 client->GetComputationShape(computation).ConsumeValueOrDie(); in RealMain()
Ddumped_computation_to_operation_list.cc78 client->GetComputationShape(computation).ConsumeValueOrDie(); in RealMain()
Ddumped_computation_to_text.cc54 client->GetComputationShape(computation).ConsumeValueOrDie(); in RealMain()
/external/tensorflow/tensorflow/compiler/aot/
Dcompile.cc54 client->GetComputationShape(computation); in CompileXla()
/external/tensorflow/tensorflow/compiler/xla/client/
Dclient.h213 StatusOr<std::unique_ptr<ProgramShape>> GetComputationShape(
Dclient.cc444 StatusOr<std::unique_ptr<ProgramShape>> Client::GetComputationShape( in GetComputationShape() function in xla::Client
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_jit_compiled_cpu_function.cc91 client->GetComputationShape(computation)); in Compile()
Dxla_compiler_test.cc1222 client_->GetComputationShape(*result.computation)); in TEST_F()
1292 client_->GetComputationShape(*result.computation)); in TEST_F()