HomeSort by relevance Sort by last modified time
    Searched refs:GetComputationShape (Results 1 - 15 of 15) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/tests/
replay_test.cc 55 client_->GetComputationShape(computation).ConsumeValueOrDie();
57 client_->GetComputationShape(replayed).ConsumeValueOrDie();
87 client_->GetComputationShape(computation).ConsumeValueOrDie();
89 client_->GetComputationShape(replayed).ConsumeValueOrDie();
133 client_->GetComputationShape(computation).ConsumeValueOrDie();
135 client_->GetComputationShape(replayed).ConsumeValueOrDie();
  /external/tensorflow/tensorflow/compiler/xla/tools/
show_signature.cc 57 client->GetComputationShape(computation).ConsumeValueOrDie();
dumped_computation_to_operation_list.cc 81 client->GetComputationShape(computation).ConsumeValueOrDie();
dumped_computation_to_text.cc 56 client->GetComputationShape(computation).ConsumeValueOrDie();
  /external/tensorflow/tensorflow/compiler/xla/client/
computation.cc 72 TF_RETURN_IF_ERROR(parent_->GetComputationShape(&request, &response));
client.h 157 StatusOr<std::unique_ptr<ProgramShape>> GetComputationShape(
client.cc 345 StatusOr<std::unique_ptr<ProgramShape>> Client::GetComputationShape(
352 Status s = stub_->GetComputationShape(&request, &response);
computation_builder.cc 244 Status status = client_->stub()->GetComputationShape(&request, &response);
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/client/lib/
testing.cc 102 client->GetComputationShape(computation).ConsumeValueOrDie();
  /external/tensorflow/tensorflow/compiler/aot/
compile.cc 52 client->GetComputationShape(computation);
  /external/tensorflow/tensorflow/compiler/xla/
service_interface.h 72 virtual tensorflow::Status GetComputationShape(
  /external/tensorflow/tensorflow/compiler/tf2xla/
xla_jit_compiled_cpu_function.cc 124 client->GetComputationShape(computation));
xla_compiler.cc 649 auto computation_shape = client()->GetComputationShape(*result->computation);
  /external/tensorflow/tensorflow/compiler/xla/service/
service.h 200 tensorflow::Status GetComputationShape(
service.cc     [all...]

Completed in 1158 milliseconds