Searched defs:ComputationLayout (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
H A Dcomputation_layout.cc26 ComputationLayout::ComputationLayout(const ProgramShape& program_shape) function in class:xla::ComputationLayout
34 void ComputationLayout::SetToDefaultLayout() {
41 bool ComputationLayout::LayoutIsSet() const {
47 string ComputationLayout::ToString() const {
H A Dcomputation_layout.h33 class ComputationLayout { class in namespace:xla
35 // Constructs a ComputationLayout from a ProgramShape. The layouts of the
38 explicit ComputationLayout(const ProgramShape& program_shape);

Completed in 105 milliseconds