Searched defs:global_shape (Results 1 – 13 of 13) sorted by relevance
42 absl::Span<const int64_t> global_shape, Layout layout) { in BuildSliceSpecDeviceMap()166 absl::Span<const int64_t> global_shape) { in SliceSpecOnDevice()
102 std::vector<int64_t> global_shape = in Compute() local
317 const std::vector<int64_t> global_shape() const { in global_shape() function
975 std::vector<int64_t> global_shape; in GlobalShapeFromLocalShape() local
45 auto global_shape = layout_op.global_shape(); in ExtractGlobalInputShape() local81 auto global_shape = layout_op.global_shape(); in ExtractGlobalOutputShape() local
1181 mlir::TF::ShapeAttr global_shape = mlir::TF::ShapeAttr::get( in InsertRelayoutForWhileLoops() local
75 const auto global_shape = result_layout_op.global_shape(); in ExpandOp() local
291 llvm::SmallVector<int64_t, 4> global_shape; in HandleConvBackpropInput() local
187 StatusOr<Layout> GetBroadcastedLayout(llvm::ArrayRef<int64_t> global_shape, in GetBroadcastedLayout()
578 std::vector<int64_t> global_shape = std::get<0>(it.value()); in ExpandRestoreV2OpHelper() local
55 llvm::ArrayRef<int64_t>* global_shape = nullptr) { in VerifySliceLayout()
1198 auto global_shape = builder.create<mlir::TF::MulOp>( in ExpandOp() local
41 global_shape, argument