Searched refs:LayoutConstraints (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | layout_assignment.h | 130 class LayoutConstraints {
132 LayoutConstraints(const TuplePointsToAnalysis& points_to_analysis,
134 ~LayoutConstraints() = default;
332 LayoutConstraints* constraints);
335 LayoutConstraints* constraints);
338 LayoutConstraints* constraints);
349 LayoutConstraints* constraints);
356 LayoutConstraints* constraints);
382 LayoutConstraints* constraints);
388 virtual Status AddBackendConstraints(LayoutConstraints* constraints) { in AddBackendConstraints()
[all …]
|
| D | layout_assignment.cc | 103 LayoutConstraints::LayoutConstraints( in LayoutConstraints() function in xla::LayoutConstraints
123 PointsToSet::BufferSet* LayoutConstraints::GetBufferSet( in GetBufferSet()
142 bool LayoutConstraints::OperandBufferForwarded( in OperandBufferForwarded()
154 Status LayoutConstraints::SetBufferLayout(const Layout& layout, in SetBufferLayout()
198 Status LayoutConstraints::SetOperandLayout(const Shape& shape_with_layout, in SetOperandLayout()
251 Status LayoutConstraints::SetArrayOperandLayout( in SetArrayOperandLayout()
262 Status LayoutConstraints::SetResultLayout(const Shape& shape_with_layout, in SetResultLayout()
287 Status LayoutConstraints::SetInstructionLayout( in SetInstructionLayout()
321 const Layout* LayoutConstraints::BufferLayout( in BufferLayout()
329 const BufferLayoutConstraint* LayoutConstraints::GetBufferLayoutConstraint( in GetBufferLayoutConstraint()
[all …]
|
| D | layout_assignment_test.cc | 523 LayoutConstraints* constraints) override { in PropagateBufferConstraint()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_layout_assignment.h | 42 Status AddBackendConstraints(LayoutConstraints* constraints) override;
45 LayoutConstraints* constraints) override;
48 LayoutConstraints* constraints) override;
52 HloCustomCallInstruction* instr, LayoutConstraints* constraints);
|
| D | gpu_layout_assignment.cc | 100 HloCustomCallInstruction* instr, LayoutConstraints* constraints) { in AddBackendConstraintsToDnnConvCustomCall()
174 LayoutConstraints* constraints) { in AddBackendConstraints()
276 LayoutConstraints* constraints) { in PropagateOperandConstraint()
329 LayoutConstraints* constraints) { in PropagateBufferConstraint()
|
| /external/deqp-deps/SPIRV-Tools/source/val/ |
| D | validate_decorations.cpp | 57 struct LayoutConstraints { struct
58 explicit LayoutConstraints( in LayoutConstraints() function
68 LayoutConstraints, PairHash>;
163 const LayoutConstraints& inherited, in getBaseAlignment()
267 uint32_t getSize(uint32_t member_id, const LayoutConstraints& inherited, in getSize()
348 const LayoutConstraints& inherited, in hasImproperStraddle()
441 const LayoutConstraints& constraint = in checkLayout()
715 const LayoutConstraints& inherited,
722 const LayoutConstraints& inherited, in ComputeMemberConstraintsForStruct()
728 LayoutConstraints& constraint = in ComputeMemberConstraintsForStruct()
[all …]
|
| /external/swiftshader/third_party/SPIRV-Tools/source/val/ |
| D | validate_decorations.cpp | 57 struct LayoutConstraints { struct
58 explicit LayoutConstraints( in LayoutConstraints() argument
68 LayoutConstraints, PairHash>;
163 const LayoutConstraints& inherited, in getBaseAlignment()
272 uint32_t getSize(uint32_t member_id, const LayoutConstraints& inherited, in getSize()
355 const LayoutConstraints& inherited, in hasImproperStraddle()
448 const LayoutConstraints& constraint = in checkLayout()
722 const LayoutConstraints& inherited,
729 const LayoutConstraints& inherited, in ComputeMemberConstraintsForStruct()
735 LayoutConstraints& constraint = in ComputeMemberConstraintsForStruct()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_layout_assignment.h | 42 Status AddBackendConstraints(LayoutConstraints* constraints) override;
|
| D | cpu_layout_assignment.cc | 110 LayoutConstraints* constraints) { in AddBackendConstraints()
|