OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:ShapeLayout
(Results
1 - 10
of
10
) sorted by null
/external/tensorflow/tensorflow/compiler/xla/
shape_layout.cc
25
tensorflow::Status
ShapeLayout
::CopyLayoutFromShape(const Shape& other_shape) {
35
tensorflow::Status
ShapeLayout
::AssignLayoutToShape(Shape* to_shape) const {
45
void
ShapeLayout
::SetToDefaultLayout() {
49
bool
ShapeLayout
::MatchesLayoutInShape(const Shape& shape) const {
53
const Layout&
ShapeLayout
::layout() const {
59
void
ShapeLayout
::Clear() { LayoutUtil::ClearLayout(&shape_); }
61
bool
ShapeLayout
::LayoutIsSet() const { return LayoutUtil::HasLayout(shape_); }
63
void
ShapeLayout
::ResetLayout(const Layout& layout) {
70
bool
ShapeLayout
::operator==(const
ShapeLayout
& other) const
[
all
...]
shape_layout.h
28
// A
ShapeLayout
object encapsulates the layout of a particular shape (including
30
// single array.
ShapeLayout
contains a Layout proto for each array in the shape
32
// trivially holds a single Layout. Logically,
ShapeLayout
holds a nonmutable
34
class
ShapeLayout
{
36
// Constructs a
ShapeLayout
of the given shape. Layouts are copied from the
38
explicit
ShapeLayout
(const Shape& shape) : shape_(shape) {}
40
// Assigns the layouts in this
ShapeLayout
to the Layout fields of the given
41
// shape. 'to_shape' and the shape of the
ShapeLayout
object must be
45
// Returns true if the Layouts in this
ShapeLayout
match the layouts in the
47
// with the
ShapeLayout
's shape, then false is returned
[
all
...]
/external/tensorflow/tensorflow/compiler/xla/service/
computation_layout.h
41
const
ShapeLayout
& parameter_layout(int64 param_no) const {
44
ShapeLayout
* mutable_parameter_layout(int64 param_no) {
52
const std::vector<
ShapeLayout
>& parameter_layouts() const {
56
// Returns the
ShapeLayout
of a result of the computation.
57
const
ShapeLayout
& result_layout() const { return result_layout_; }
58
ShapeLayout
* mutable_result_layout() { return &result_layout_; }
77
std::vector<
ShapeLayout
> parameter_layouts_;
78
ShapeLayout
result_layout_;
layout_assignment.h
92
OperandLayoutConstraint(const
ShapeLayout
& shape_layout,
96
const
ShapeLayout
& shape_layout() const { return shape_layout_; }
106
ShapeLayout
shape_layout_;
114
explicit ResultLayoutConstraint(const
ShapeLayout
& shape_layout,
119
const
ShapeLayout
& shape_layout() const { return shape_layout_; }
123
const
ShapeLayout
shape_layout_;
157
const
ShapeLayout
* OperandLayout(const HloInstruction* instruction,
161
const
ShapeLayout
* ResultLayout() const;
335
Status PropagateUseConstraintToDefs(const
ShapeLayout
& shape_layout,
computation_layout.cc
43
[](const
ShapeLayout
& s) { return s.LayoutIsSet(); }) &&
hlo_module_config.cc
46
for (const
ShapeLayout
& param_layout :
layout_assignment_test.cc
80
const
ShapeLayout
shape_layout(shape);
110
const
ShapeLayout
col_major(col_major_shape);
115
const
ShapeLayout
row_major(row_major_shape);
162
const
ShapeLayout
shape_layout(shape);
345
ShapeLayout
(ashape_with_layout);
346
*computation_layout.mutable_result_layout() =
ShapeLayout
(bshape_with_layout);
386
ShapeLayout
(ashape_with_layout);
387
*computation_layout.mutable_result_layout() =
ShapeLayout
(bshape_with_layout);
424
ShapeLayout
(input_shape_with_layout);
426
ShapeLayout
(output_shape_with_layout)
[
all
...]
layout_assignment.cc
113
Status CopyOperandIfLayoutsDiffer(const
ShapeLayout
& operand_layout,
153
const
ShapeLayout
& shape_layout, const HloInstruction* instruction,
307
key, OperandLayoutConstraint(
ShapeLayout
(shape_with_layout),
312
OperandLayoutConstraint(
ShapeLayout
(shape_with_layout), instruction,
336
const
ShapeLayout
* curr_shape_layout = ResultLayout();
350
new ResultLayoutConstraint(
ShapeLayout
(shape_with_layout), dfs));
405
const
ShapeLayout
* LayoutConstraints::OperandLayout(
420
const
ShapeLayout
* LayoutConstraints::ResultLayout() const {
[
all
...]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
gpu_layout_assignment_test.cc
66
ShapeLayout
(lhs_shape_with_layout);
68
ShapeLayout
(rhs_shape_with_layout);
70
ShapeLayout
(result_shape_with_layout);
152
ShapeLayout
(input_shape);
156
*computation_layout.mutable_result_layout() =
ShapeLayout
(result_shape);
219
ShapeLayout
(input_shape);
224
ShapeLayout
(ShapeUtil::MakeTupleShape(
299
ShapeLayout
(input_shape);
304
ShapeLayout
(ShapeUtil::MakeTupleShape(
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_layout_assignment_test.cc
74
ShapeLayout
(LayoutUtil::GetWithDefaultLayout(lhs_shape));
76
ShapeLayout
(LayoutUtil::GetWithDefaultLayout(result_shape));
115
ShapeLayout
(LayoutUtil::GetWithDefaultLayout(lhs_shape));
117
ShapeLayout
(LayoutUtil::GetWithDefaultLayout(result_shape));
159
ShapeLayout
(LayoutUtil::GetWithDefaultLayout(lhs_a_shape));
161
ShapeLayout
(LayoutUtil::GetWithDefaultLayout(lhs_b_shape));
163
ShapeLayout
(LayoutUtil::GetWithDefaultLayout(tuple_result->shape()));
193
ShapeLayout
(LayoutUtil::GetWithDefaultLayout(rhs_shape));
195
ShapeLayout
(LayoutUtil::GetWithDefaultLayout(result_shape));
233
ShapeLayout
(LayoutUtil::GetWithDefaultLayout(lhs_shape))
[
all
...]
Completed in 636 milliseconds