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

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
gpu_layout_assignment.h 40 const BufferLayoutConstraint& buffer_constraint,
gpu_layout_assignment.cc 208 const BufferLayoutConstraint& buffer_constraint,
  /external/tensorflow/tensorflow/compiler/xla/service/
layout_assignment.h 70 class BufferLayoutConstraint : public LayoutConstraint {
72 BufferLayoutConstraint(const Layout& layout, const LogicalBuffer& buffer,
155 const BufferLayoutConstraint* GetBufferLayoutConstraint(
203 std::unordered_map<const LogicalBuffer*, BufferLayoutConstraint>
299 const BufferLayoutConstraint& layout_constraint,
329 const BufferLayoutConstraint& layout_constraint,
layout_assignment.cc 139 BufferLayoutConstraint::BufferLayoutConstraint(const Layout& layout,
146 string BufferLayoutConstraint::ToString() const {
147 return tensorflow::strings::Printf("BufferLayoutConstraint %s: %s",
229 const BufferLayoutConstraint* curr_constraint =
252 BufferLayoutConstraint(layout, buffer, mandatory, dfs)))
255 iter->second = BufferLayoutConstraint(layout, buffer, mandatory, dfs);
370 // Create a BufferLayoutConstraint for each array shape in the output of the
399 const BufferLayoutConstraint* LayoutConstraints::GetBufferLayoutConstraint(
    [all...]
layout_assignment_test.cc 495 const BufferLayoutConstraint& buffer_constraint,
    [all...]

Completed in 95 milliseconds