HomeSort by relevance Sort by last modified time
    Searched refs:AssignLayouts (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
layout_assignment_test.cc 54 void AssignLayouts(HloModule* m, ComputationLayout* entry_computation_layout,
115 AssignLayouts(m.get(), &computation_layout);
151 AssignLayouts(m.get(), &computation_layout);
196 AssignLayouts(m.get(), &computation_layout);
237 AssignLayouts(m.get(), &computation_layout);
279 AssignLayouts(m.get(), &computation_layout);
318 AssignLayouts(m.get(), &computation_layout);
375 AssignLayouts(m.get(), &computation_layout);
416 AssignLayouts(m.get(), &computation_layout);
455 AssignLayouts(m.get(), &computation_layout)
    [all...]
layout_assignment.h 407 Status AssignLayouts(const LayoutConstraints& constraints,
layout_assignment.cc     [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_layout_assignment_test.cc 51 void AssignLayouts(HloModule* module,
84 AssignLayouts(module.get(), &computation_layout);
125 AssignLayouts(module.get(), &computation_layout);
171 AssignLayouts(module.get(), &computation_layout);
203 AssignLayouts(module.get(), &computation_layout);
243 AssignLayouts(module.get(), &computation_layout);

Completed in 95 milliseconds