Home
last modified time | relevance | path

Searched defs:xla_output_shape (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_helpers.h143 xla::Shape xla_output_shape; member
/external/tensorflow/tensorflow/core/tpu/kernels/xla/
Dhost_compute_ops.cc194 xla::Shape* xla_output_shape; in Compile() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dcompile_mlir_util.cc188 xla::Shape* xla_output_shape, std::vector<XlaOutputDescription>* outputs, in GetOutputInfo()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_executable.cc1009 absl::string_view module_name, Shape xla_output_shape, in GpuExecutable()