Home
last modified time | relevance | path

Searched defs:GetProgramShape (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dxla_computation.cc26 StatusOr<ProgramShape> XlaComputation::GetProgramShape() const { in GetProgramShape() function in xla::XlaComputation
H A Dxla_builder.cc313 StatusOr<ProgramShape> XlaBuilder::GetProgramShape(int64_t root_id) const { in GetProgramShape() function in xla::XlaBuilder
344 StatusOr<ProgramShape> XlaBuilder::GetProgramShape() const { in GetProgramShape() function in xla::XlaBuilder
349 StatusOr<ProgramShape> XlaBuilder::GetProgramShape(XlaOp root) const { in GetProgramShape() function in xla::XlaBuilder