Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_helpers.h149 xla::Shape xla_output_shape; member
/external/tensorflow/tensorflow/core/tpu/kernels/xla/
Dhost_compute_ops.cc180 xla::Shape* xla_output_shape; in Compile() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dcompile_mlir_util.cc139 xla::Shape* xla_output_shape, std::vector<XlaOutputDescription>* outputs, in GetOutputInfo()