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.cc53 client_->GetComputationShape(computation).value(); in TEST_F()
55 client_->GetComputationShape(replayed).value(); in TEST_F()
85 client_->GetComputationShape(computation).value(); in XLA_TEST_F()
87 client_->GetComputationShape(replayed).value(); in XLA_TEST_F()
129 client_->GetComputationShape(computation).value(); in TEST_F()
131 client_->GetComputationShape(replayed).value(); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/tools/
Dshow_signature.cc55 client->GetComputationShape(computation).value(); in RealMain()
Ddumped_computation_to_operation_list.cc80 client->GetComputationShape(computation).value(); in RealMain()
Ddumped_computation_to_text.cc56 client->GetComputationShape(computation).value(); in RealMain()
/external/tensorflow/tensorflow/compiler/xla/client/
Dclient.h212 StatusOr<std::unique_ptr<ProgramShape>> GetComputationShape(
Dclient.cc445 StatusOr<std::unique_ptr<ProgramShape>> Client::GetComputationShape( in GetComputationShape() function in xla::Client
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_jit_compiled_cpu_function.cc98 client->GetComputationShape(computation)); in Compile()
Dxla_compiler_test.cc1367 client_->GetComputationShape(*result.computation)); in TEST_F()
1440 client_->GetComputationShape(*result.computation)); in TEST_F()
/external/tensorflow/tensorflow/compiler/aot/
Dcompile.cc68 client->GetComputationShape(computation); in CompileXla()