| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | shape_util.cc | 50 StatusOr<bool> MakeLayout(absl::Span<const int64_t> minor_to_major, in MakeLayout()
73 absl::Span<const int64_t> minor_to_major, in AssignLayout()
173 const xla::Shape& input_shape, absl::Span<const int64_t> minor_to_major, in GetShapeWithLayout()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | layout.h | 88 explicit Layout(absl::Span<const int64_t> minor_to_major) in Layout()
190 int64_t minor_to_major(int index) const { return minor_to_major_.at(index); } in minor_to_major() function
203 absl::Span<const int64_t> minor_to_major() const { return minor_to_major_; } in minor_to_major() function
|
| D | layout_util.cc | 44 void SetDefaultLayoutToContainer(T* minor_to_major) { in SetDefaultLayoutToContainer()
57 absl::Span<const int64_t> minor_to_major, in MakeLayout()
110 auto* minor_to_major = layout.mutable_minor_to_major(); in CreateDefaultLayoutForRank() local
153 auto* minor_to_major = shape->mutable_layout()->mutable_minor_to_major(); in SetToDefaultLayout() local
|
| D | shape_util.cc | 101 absl::Span<const int64_t> minor_to_major, in MakeShapeWithLayoutInternal()
228 auto* minor_to_major = layout->mutable_minor_to_major(); in FillNewShape() local
314 absl::Span<const int64_t> minor_to_major, in MakeShapeWithLayout()
340 DimensionVector minor_to_major; in MoveDimToMajor() local
1763 auto minor_to_major = shape.layout().minor_to_major(); in ArraySize() local
|
| D | layout_util_test.cc | 31 absl::Span<const int64_t> minor_to_major, in MakeShapeWithLayout()
|
| D | literal_util.cc | 390 absl::Span<const int64_t> minor_to_major, const LiteralSlice& literal) { in ReshapeSlice()
|
| D | xla_data.proto | 156 repeated int64 minor_to_major = 1; field
|
| D | literal_test.cc | 1528 auto* minor_to_major = in SetDefaultLayoutOnProto() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | adjust_layout.cc | 61 auto minor_to_major = new_shape.layout().minor_to_major(); in GetTPUInfeedLayoutFromAPI() local
95 std::vector<int64_t> minor_to_major; in GetTPUInfeedLayout() local
|
| D | mhlo_to_lhlo_with_xla.cc | 519 llvm::SmallVector<int64_t, 4> minor_to_major( in RewriteFusionOperand() local
927 std::vector<int64_t> minor_to_major(layout.minor_to_major_size()); in EmitDnnConvolution() local
1356 llvm::SmallVector<int64_t, 4> minor_to_major(layout.minor_to_major().begin(), in GetLayoutAttribute() local
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | dlpack.cc | 197 std::vector<int64_t> minor_to_major(dims.size()); in StridesToLayout() local
374 std::vector<int64_t> minor_to_major; in DLPackManagedTensorToBuffer() local
|
| D | xla_compiler.cc | 155 std::optional<absl::Span<const int64_t>> minor_to_major, in MakeShapeWithLayout()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | type_to_shape.cc | 128 llvm::SmallVector<int64_t, 4> minor_to_major; in TypeToShape() local
|
| D | hlo_function_importer.cc | 1723 llvm::SmallVector<int64_t, 4> minor_to_major( in SetLayoutForMlir() local
1747 std::vector<mlir::Attribute> minor_to_major; in ConvertShapeToMlirLayout() local
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | infeed_ops.cc | 116 std::vector<int64_t>* minor_to_major) { in GetLayoutOverride()
128 std::vector<int64_t> minor_to_major; in GetInfeedShapeWithLayout() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | layout_assignment_test.cc | 65 auto minor_to_major = instr->shape().layout().minor_to_major(); in LayoutOf() local
70 absl::Span<const int64_t> minor_to_major) { in ExpectLayoutIs()
80 for (const absl::Span<const int64_t> minor_to_major : minor_to_majors) { in ExpectTupleLayoutIs() local
95 for (auto& minor_to_major : minor_to_majors) { in TEST_F() local
164 for (auto& minor_to_major : minor_to_majors) { in TEST_F() local
|
| D | copy_insertion_test.cc | 138 auto minor_to_major = LayoutUtil::MinorToMajor(constant->shape()); in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | matmul_utils.cc | 76 std::vector<int64_t> minor_to_major; in GetBatchRowColumnShape() local
134 absl::Span<const int64_t> minor_to_major = shape.layout().minor_to_major(); in For() local
|
| D | jitrt_custom_calls.cc | 319 llvm::SmallVector<int64_t> minor_to_major; in ToShape() local
728 ArrayRef<int64_t> minor_to_major) { in GetConvDescriptor()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | matrix_ops_simple_test.cc | 189 auto minor_to_major = [](bool row_major) -> std::vector<int64_t> { in TestImpl() local
|
| D | client_library_test_base.cc | 195 std::vector<int64_t> minor_to_major(expected.shape().rank()); in ComputeAndCompareLiteralWithAllOutputLayouts() local
238 std::vector<int64_t> minor_to_major(literal.shape().rank()); in ComputeAndCompareLiteralWithAllInputLayouts() local
|
| D | broadcast_simple_test.cc | 58 absl::Span<const int64_t> minor_to_major, Shape* r3_shape, in MakeR3Data()
71 absl::Span<const int64_t> minor_to_major, Shape* r2_shape, in MakeR2Data()
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/ |
| D | c_api_conversions.cc | 300 absl::Span<const int64_t> minor_to_major = MakeSpan(c_layout->minor_to_major); in FromC() local
|
| D | c_api_decl.h | 237 Int64List minor_to_major; member
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_state_ops.h | 269 std::vector<int64_t> minor_to_major; in XRTAllocateFromTensorOp() local
|