Home
last modified time | relevance | path

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

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