Searched refs:LayoutConstraints (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | layout_assignment.h | 132 class LayoutConstraints {
134 LayoutConstraints(const TuplePointsToAnalysis& points_to_analysis,
136 ~LayoutConstraints() = default;
334 LayoutConstraints* constraints);
337 LayoutConstraints* constraints);
340 LayoutConstraints* constraints);
354 LayoutConstraints* constraints);
361 LayoutConstraints* constraints);
391 LayoutConstraints* constraints);
397 virtual Status AddBackendConstraints(LayoutConstraints* constraints) { in AddBackendConstraints()
[all …]
|
| D | layout_assignment.cc | 108 LayoutConstraints::LayoutConstraints( in LayoutConstraints() function in xla::LayoutConstraints
128 PointsToSet::BufferSet* LayoutConstraints::GetBufferSet( in GetBufferSet()
147 bool LayoutConstraints::OperandBufferForwarded( in OperandBufferForwarded()
159 Status LayoutConstraints::SetBufferLayout(const Layout& layout, in SetBufferLayout()
208 Status LayoutConstraints::SetOperandLayout(const Shape& shape_with_layout, in SetOperandLayout()
261 Status LayoutConstraints::SetArrayOperandLayout( in SetArrayOperandLayout()
272 Status LayoutConstraints::SetResultLayout(const Shape& shape_with_layout, in SetResultLayout()
297 Status LayoutConstraints::SetInstructionLayout( in SetInstructionLayout()
331 const Layout* LayoutConstraints::BufferLayout( in BufferLayout()
339 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 | 122 HloCustomCallInstruction* instr, LayoutConstraints* constraints) { in AddBackendConstraintsToDnnConvCustomCall()
196 LayoutConstraints* constraints) { in AddBackendConstraints()
309 LayoutConstraints* constraints) { in PropagateOperandConstraint()
362 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()
272 uint32_t getSize(uint32_t member_id, const LayoutConstraints& inherited, in getSize()
355 const LayoutConstraints& inherited, in hasImproperStraddle()
454 const LayoutConstraints& constraint = in checkLayout()
828 const LayoutConstraints& inherited,
835 const LayoutConstraints& inherited, in ComputeMemberConstraintsForStruct()
841 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()
454 const LayoutConstraints& constraint = in checkLayout()
789 const LayoutConstraints& inherited,
796 const LayoutConstraints& inherited, in ComputeMemberConstraintsForStruct()
802 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()
|