/external/tensorflow/tensorflow/compiler/tf2xla/ |
D | shape_util.cc | 50 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/ |
D | layout.h | 82 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
|
D | layout_util.cc | 44 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
|
D | literal_util.cc | 294 absl::Span<const int64> minor_to_major, const LiteralSlice& literal) { in ReshapeSlice()
|
D | layout_util_test.cc | 31 absl::Span<const int64> minor_to_major) { in MakeShapeWithLayout()
|
D | shape_util.cc | 93 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()
|
D | xla_data.proto | 143 repeated int64 minor_to_major = 1; field
|
D | literal_test.cc | 1451 auto* minor_to_major = in SetDefaultLayoutOnProto() local
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | layout_assignment_test.cc | 63 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
|
D | copy_insertion_test.cc | 136 auto minor_to_major = LayoutUtil::MinorToMajor(constant->shape()); in TEST_F() local
|
D | hlo_parser.cc | 3450 std::vector<int64> minor_to_major; in ParseLayout() local
|
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
D | kernel_tiling.cc | 62 auto minor_to_major = shape.layout().minor_to_major(); in GetReshapedIndex() local
|
/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | matrix_ops_simple_test.cc | 191 auto minor_to_major = [](bool row_major) -> std::vector<int64> { in TestImpl() local
|
D | client_library_test_base.cc | 195 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
|
D | broadcast_simple_test.cc | 57 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/ |
D | xrt_state_ops.h | 220 std::vector<int64> minor_to_major; in XRTAllocateFromTensorOp() local
|
/external/tensorflow/tensorflow/compiler/xla/python/ |
D | numpy_bridge.cc | 278 std::vector<int64> minor_to_major(length); in XlaShapeFromPyShape() local
|
D | xla_client.py | 486 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/ |
D | ir_emitter_unnested.cc | 3536 absl::Span<const int64> minor_to_major = in GetReductionToVectorDimensions() local
|