Home
last modified time | relevance | path

Searched refs:Replicate (Results 1 – 25 of 55) sorted by relevance

123

/external/eigen/Eigen/src/Core/
DReplicate.h17 struct traits<Replicate<MatrixType,RowFactor,ColFactor> >
61 template<typename MatrixType,int RowFactor,int ColFactor> class Replicate
62 : public internal::dense_xpr_base< Replicate<MatrixType,RowFactor,ColFactor> >::type
64 typedef typename internal::traits<Replicate>::MatrixTypeNested MatrixTypeNested;
65 typedef typename internal::traits<Replicate>::_MatrixTypeNested _MatrixTypeNested;
68 typedef typename internal::dense_xpr_base<Replicate>::type Base;
69 EIGEN_DENSE_PUBLIC_INTERFACE(Replicate)
74 inline explicit Replicate(const OriginalMatrixType& matrix)
84 inline Replicate(const OriginalMatrixType& matrix, Index rowFactor, Index colFactor)
118 const Replicate<Derived,RowFactor,ColFactor>
[all …]
DVectorwiseOp.h208 typedef Replicate<OtherDerived,
231 typedef Replicate<OtherDerived,
471 …typedef Replicate<ExpressionType,(isVertical?Dynamic:1),(isHorizontal?Dynamic:1)> ReplicateReturnT…
485 …template<int Factor> const Replicate<ExpressionType,isVertical*Factor+isHorizontal,isHorizontal*Fa…
489 return Replicate<ExpressionType,(isVertical?Factor:1),(isHorizontal?Factor:1)>
653 const Replicate<HNormalized_Factors,
DDenseBase.h543 const Replicate<Derived,RowFactor,ColFactor> replicate() const;
554 const Replicate<Derived, Dynamic, Dynamic> replicate(Index rowFactor, Index colFactor) const in replicate()
556 return Replicate<Derived, Dynamic, Dynamic>(derived(), rowFactor, colFactor); in replicate()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_test.cc56 TEST_F(HloShardingTest, Replicate) { in TEST_F() argument
57 HloSharding sharding = HloSharding::Replicate(); in TEST_F()
63 HloSharding other = HloSharding::Replicate(); in TEST_F()
79 HloSharding other = HloSharding::Replicate(); in TEST_F()
171 *proto.add_tuple_shardings() = HloSharding::Replicate().ToProto(); in TEST_F()
179 EXPECT_EQ(shape_tree.element({0}), HloSharding::Replicate()); in TEST_F()
201 HloSharding sharding1 = HloSharding::Replicate(); in TEST_F()
202 HloSharding sharding2 = HloSharding::Replicate(); in TEST_F()
224 HloSharding default_sharding = HloSharding::Replicate(); in TEST_F()
228 HloSharding sharding1 = HloSharding::Replicate(); in TEST_F()
[all …]
Dhlo_sharding_util_test.cc25 EXPECT_EQ(TransposeSharding(HloSharding::Replicate(), {0, 1, 2}), in TEST()
26 HloSharding::Replicate()); in TEST()
Dhlo_sharding.cc53 return Replicate(metadata); in PartialTile()
73 return Replicate(metadata); in PartialTile()
327 ShapeTree<HloSharding> result(shape, HloSharding::Replicate()); in AsShapeTree()
489 return Replicate(metadata); in FromProto()
Dbatchnorm_expander.cc279 : HloSharding::Replicate(); in HandleBatchNormTraining()
366 : HloSharding::Replicate(); in HandleBatchNormInference()
543 : HloSharding::Replicate(); in HandleBatchNormGrad()
Dsharding_propagation.cc122 HloSharding::SingleTuple(instruction->shape(), HloSharding::Replicate())); in SetDefaultTupleSharding()
538 HloSharding::Replicate(lhs->sharding().metadata()), instruction, in InferConvolutionShardingFromOperands()
555 return MaybeImproveInstructionSharding(HloSharding::Replicate(), instruction, in InferConvolutionShardingFromOperands()
613 HloSharding::Replicate(op->sharding().metadata()), instruction, in InferShardingFromOperands()
711 HloSharding::Replicate(operand->sharding().metadata())), in InferShardingFromOperands()
921 HloSharding::Replicate(operand->sharding().metadata()), in InferShardingFromOperands()
1008 HloSharding::Replicate(), instruction, may_combine_partial_sharding); in InferShardingFromOperands()
1281 HloSharding::Replicate()); in GetShardingFromUser()
Dhlo_sharding_util.cc527 return HloSharding::Replicate(index_sharding.metadata()); in GatherOutputSharding()
574 return HloSharding::Replicate(output_sharding.metadata()); in GatherIndexSharding()
663 return HloSharding::Replicate(data_sharding.metadata()); in ScatterIndexSharding()
696 return HloSharding::Replicate(index_sharding.metadata()); in ScatterDataSharding()
1158 return HloSharding::Replicate(sharding.metadata()); in PartiallyReplicateTiledShardingOnDims()
Dhlo_sharding.h45 static HloSharding Replicate(absl::Span<const OpMetadata> metadata = {}) {
Dhlo_matchers_test.cc176 HloSharding::Replicate()}); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dgather_scatter_handler.cc145 auto replicated_operand = operand.Replicate(); in PartitionIndexOnlyPartition()
190 indices = indices.Reshard(HloSharding::Replicate()); in ParititonPassthroughOperand()
226 indices = indices.Reshard(HloSharding::Replicate()); in ParititonTrivialIndexedOperandDimension()
302 ar->set_sharding(HloSharding::Replicate()); in ParititonTrivialIndexedOperandDimension()
598 identity, operand.Replicate().hlo())); in HandleScatter()
608 all_reduce->set_sharding(HloSharding::Replicate()); in HandleScatter()
621 indices = indices.Reshard(HloSharding::Replicate()); in HandleScatter()
644 indices = indices.Reshard(HloSharding::Replicate()); in HandleScatter()
645 updates = updates.Reshard(HloSharding::Replicate()); in HandleScatter()
Dspmd_partitioner.cc443 return Replicate().Reshard(target); in ReshardNoCache()
814 return Replicate().ReshardAsWindowedInput(window, target, pad_value); in ReshardAsWindowedInput()
823 PartitionedHlo PartitionedHlo::Replicate() { in Replicate() function in xla::spmd::PartitionedHlo
848 cache.emplace_back(HloSharding::Replicate(), std::move(resharded)); in Replicate()
862 result->set_sharding(HloSharding::Replicate()); in Replicate()
1082 result->set_sharding(HloSharding::Replicate()); in Broadcast()
1389 : HloSharding::Replicate(); in DefaultAction()
1712 partitioned_slice_input.Reshard(HloSharding::Replicate()).hlo(); in HandleSort()
1722 partitioned_slice_index.Reshard(HloSharding::Replicate()).hlo(); in HandleSort()
1733 final_sort->set_sharding(HloSharding::Replicate() in HandleSort()
[all …]
Ddot_handler.cc826 padded_slice_operand->set_sharding(HloSharding::Replicate()); in PartitionBaseCase()
1078 slice_operand->set_sharding(HloSharding::Replicate()); in PartitionBaseCase()
1522 ar->set_sharding(HloSharding::Replicate()); in PartitionBaseCase()
1532 auto rhs_replicated = rhs.Reshard(HloSharding::Replicate()).hlo(); in PartitionBaseCase()
1542 auto lhs_replicated = lhs.Reshard(HloSharding::Replicate()).hlo(); in PartitionBaseCase()
1564 auto replicated_rhs = rhs.Reshard(HloSharding::Replicate()); in PartitionBaseCase()
1572 auto replicated_lhs = lhs.Reshard(HloSharding::Replicate()); in PartitionBaseCase()
1755 operand.Replicate().hlo(), operand.state().partition_id, in PartitionDotGroupOnBatch()
1758 partially_sharded->set_sharding(HloSharding::Replicate()); in PartitionDotGroupOnBatch()
2030 other = other.Replicate(); in PartitionDotGroupOnNonContracting()
[all …]
Dconvolution_handler.cc505 ar->set_sharding(HloSharding::Replicate()); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
732 ar->set_sharding(HloSharding::Replicate()); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
771 rhs = rhs.Reshard(HloSharding::Replicate()); in PartitionConvolutionTiledOutput()
Dspmd_partitioner.h338 PartitionedHlo Replicate();
/external/tensorflow/tensorflow/compiler/xla/client/
Dsharding_builder.h34 OpSharding Replicate();
Dsharding_builder.cc21 OpSharding Replicate() { in Replicate() function
/external/tensorflow/tensorflow/compiler/tf2xla/
Dsharding_util_test.cc138 ::testing::Values(xla::sharding_builder::Replicate(),
/external/eigen/Eigen/src/Geometry/
DHomogeneous.h201 Replicate<HNormalized_Factors,
403 typedef Replicate<const ConstantColumn,Rows,1> ConstantBlock;
456 typedef Replicate<const ConstantColumn,1,Cols> ConstantBlock;
/external/eigen/unsupported/Eigen/src/Splines/
DSpline.h305 const Replicate<BasisVectorType,Dimension,1> ctrl_weights(basis_funcs); in operator()
336 … const Replicate<BasisDerivativeRowXpr,Dimension,1> ctrl_weights( basis_func_ders.row(der_order) ); in derivativesImpl()
/external/oss-fuzz/projects/proxygen/
DDockerfile132 # Replicate `install-dependencies` from the proxygen `build.sh` script
/external/llvm-project/lld/test/COFF/
Dpdb-type-server-simple.test1 Replicate this scenario:
/external/tensorflow/tensorflow/compiler/xla/
Dliteral.h342 Literal Replicate(int64 times) const;
1122 Literal LiteralBase::Replicate(int64 times) const { in Replicate() function
/external/tensorflow/tensorflow/python/ops/numpy_ops/g3doc/
DTensorFlow_Numpy_Distributed_Image_Classification.ipynb513 "# Replicate and run the parameter initialization.\n",
516 "# Replicate the training step\n",

123