Home
last modified time | relevance | path

Searched defs:minor_to_major (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dshape_util.cc50 xla::StatusOr<bool> MakeLayout(absl::Span<const int64> minor_to_major, in MakeLayout()
72 absl::Span<const int64> minor_to_major, in AssignLayout()
131 const xla::Shape& input_shape, absl::Span<const int64> minor_to_major, in GetShapeWithLayout()
/external/tensorflow/tensorflow/compiler/xla/
Dlayout.h82 explicit Layout(absl::Span<const int64> minor_to_major) in Layout()
154 int64 minor_to_major(int index) const { return minor_to_major_.at(index); } in minor_to_major() function
167 const std::vector<int64>& minor_to_major() const { return minor_to_major_; } in minor_to_major() function
Dlayout_util.cc44 void SetDefaultLayoutToContainer(std::vector<int64>* minor_to_major) { in SetDefaultLayoutToContainer()
57 absl::Span<const int64> minor_to_major, absl::Span<const Tile> tiles, in MakeLayout()
107 std::vector<int64>* minor_to_major = layout.mutable_minor_to_major(); in CreateDefaultLayoutForRank() local
151 auto* minor_to_major = shape->mutable_layout()->mutable_minor_to_major(); in SetToDefaultLayout() local
458 for (int64 minor_to_major : layout.minor_to_major()) { in Hash() local
Dliteral_util.cc294 absl::Span<const int64> minor_to_major, const LiteralSlice& literal) { in ReshapeSlice()
Dlayout_util_test.cc31 absl::Span<const int64> minor_to_major) { in MakeShapeWithLayout()
Dshape_util.cc93 absl::Span<const int64> minor_to_major, absl::Span<const Tile> tiles, in MakeShapeWithLayoutInternal()
191 absl::Span<const int64> minor_to_major, absl::Span<const Tile> tiles, in MakeShapeWithLayout()
Dxla_data.proto143 repeated int64 minor_to_major = 1; field
Dliteral_test.cc1451 auto* minor_to_major = in SetDefaultLayoutOnProto() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment_test.cc63 auto minor_to_major = in LayoutOf() local
69 absl::Span<const int64> minor_to_major) { in ExpectLayoutIs()
79 for (const absl::Span<const int64> minor_to_major : minor_to_majors) { in ExpectTupleLayoutIs() local
94 for (auto& minor_to_major : minor_to_majors) { in TEST_F() local
163 for (auto& minor_to_major : minor_to_majors) { in TEST_F() local
Dcopy_insertion_test.cc136 auto minor_to_major = LayoutUtil::MinorToMajor(constant->shape()); in TEST_F() local
Dhlo_parser.cc3450 std::vector<int64> minor_to_major; in ParseLayout() local
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dkernel_tiling.cc62 auto minor_to_major = shape.layout().minor_to_major(); in GetReshapedIndex() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dmatrix_ops_simple_test.cc191 auto minor_to_major = [](bool row_major) -> std::vector<int64> { in TestImpl() local
Dclient_library_test_base.cc195 std::vector<int64> minor_to_major(expected.shape().rank()); in ComputeAndCompareLiteralWithAllOutputLayouts() local
238 std::vector<int64> minor_to_major(literal.shape().rank()); in ComputeAndCompareLiteralWithAllInputLayouts() local
Dbroadcast_simple_test.cc57 absl::Span<const int64> minor_to_major, in MakeR3Data()
71 absl::Span<const int64> minor_to_major, in MakeR2Data()
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_state_ops.h220 std::vector<int64> minor_to_major; in XRTAllocateFromTensorOp() local
/external/tensorflow/tensorflow/compiler/xla/python/
Dnumpy_bridge.cc278 std::vector<int64> minor_to_major(length); in XlaShapeFromPyShape() local
Dxla_client.py486 def array_shape(element_type, dimensions, minor_to_major=None): argument
504 def __init__(self, dimensions, dtype, minor_to_major=None): argument
564 def minor_to_major(self): member in Shape
592 def update_minor_to_major(self, minor_to_major): argument
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc3536 absl::Span<const int64> minor_to_major = in GetReductionToVectorDimensions() local