Home
last modified time | relevance | path

Searched refs:Dimensions (Results 1 – 25 of 96) sorted by relevance

1234

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/
Drec_generate_strategy.h36 Strategys PrepareBiasAdd(const std::shared_ptr<Dimensions> &s);
38 Dimensions basic_stra);
44 …therV2(const std::vector<std::shared_ptr<OperatorInfo>> &ops, const size_t iter_ops, Dimensions s);
45 …herV2P(const std::vector<std::shared_ptr<OperatorInfo>> &ops, const size_t iter_ops, Dimensions s);
46 Dimensions PrepareGatherV2POutputStrategy(const std::vector<std::shared_ptr<OperatorInfo>> &ops,
49 Dimensions s);
53 …adcast(const std::vector<std::shared_ptr<OperatorInfo>> &ops, const size_t iter_ops, Dimensions s);
54 Dimensions ApplyBroadcast(const std::vector<std::shared_ptr<OperatorInfo>> &ops, const size_t iter_…
56 …isible(const std::vector<std::shared_ptr<OperatorInfo>> &ops, const size_t iter_ops, Dimensions s);
71 Dimensions CopyIncomingOperatorOutputStrategy(const std::shared_ptr<Graph> &graph,
[all …]
Drec_generate_strategy.cc125 Dimensions s; in PrepareMatMul()
147 Strategys PrepareBiasAdd(const std::shared_ptr<Dimensions> &s) { in PrepareBiasAdd()
150 Dimensions s_biasadd; in PrepareBiasAdd()
157 Dimensions basic_stra) { in PrepareStridedSlice()
202 Dimensions s_empty = {}; in PrepareOneHot()
208 …herV2(const std::vector<std::shared_ptr<OperatorInfo>> &ops, const size_t iter_ops, Dimensions s) { in PrepareGatherV2()
225 …erV2P(const std::vector<std::shared_ptr<OperatorInfo>> &ops, const size_t iter_ops, Dimensions s) { in PrepareGatherV2P()
229 Dimensions index(output_shape.size() - 1, 0); in PrepareGatherV2P()
239 Dimensions strategie(output_shape.size(), 1); in PrepareGatherV2P()
291 Dimensions PrepareGatherV2POutputStrategy(const std::vector<std::shared_ptr<OperatorInfo>> &ops, in PrepareGatherV2POutputStrategy()
[all …]
/third_party/mindspore/tests/ut/cpp/parallel/ops_info/
Dgenerate_strategy_test.cc61 Dimensions input0_strategy = sp->GetInputDim()[0]; in TEST_F()
62 Dimensions input1_strategy = sp->GetInputDim()[1]; in TEST_F()
74 Dimensions input0_strategy = sp->GetInputDim()[0]; in TEST_F()
75 Dimensions input1_strategy = sp->GetInputDim()[1]; in TEST_F()
89 Dimensions input0_strategy = sp->GetInputDim()[0]; in TEST_F()
90 Dimensions input1_strategy = sp->GetInputDim()[1]; in TEST_F()
104 Dimensions input0_strategy = sp->GetInputDim()[0]; in TEST_F()
105 Dimensions input1_strategy = sp->GetInputDim()[1]; in TEST_F()
118 Dimensions input0_strategy = sp->GetInputDim()[0]; in TEST_F()
119 Dimensions input1_strategy = sp->GetInputDim()[1]; in TEST_F()
[all …]
/third_party/boost/boost/units/detail/
Dheterogeneous_conversion.hpp197 template<class Begin, class Current, class Units, class Dimensions, int N>
203 Dimensions
210 template<class Begin, class Current, class Units, class Dimensions, int N>
218 template<class Begin, class Current, class Units, class Dimensions, int N>
224 Dimensions
231 template<class Begin, class Current, class Units, class Dimensions, int N>
237 Dimensions
244 template<class Begin, class Current, class Units, class Dimensions>
246 typedef typename expand_dimensions<Dimensions::size::value>::template apply<
247 Dimensions,
[all …]
Dlinear_algebra.hpp806 template<class Begin, class Dimensions>
808 …typedef typename create_unit_matrix<N - 1>::template apply<typename Begin::next, Dimensions>::type…
809 …typedef list<typename expand_dimensions<Dimensions::size::value>::template apply<Dimensions, typen…
815 template<class Begin, class Dimensions>
1012 template<class T, class Dimensions>
1016 Dimensions
1023 template<class T, class Dimensions>
1034 Dimensions
1049 template<class T, class Dimensions>
1051 …ate_base_unit_exponents_impl<is_simple_system<T>::value>::template apply<T, Dimensions>::type type;
/third_party/mindspore/tests/ut/cpp/parallel/
Dstrategy_test.cc36 Dimensions dimension1 = {2, 4}; in TEST_F()
37 Dimensions dimension2 = {2, 2}; in TEST_F()
47 Dimensions dimension1 = {2, 4}; in TEST_F()
48 Dimensions dimension2 = {2, 2}; in TEST_F()
58 Dimensions dimension1 = {2, 4}; in TEST_F()
59 Dimensions dimension2 = {2, 2}; in TEST_F()
69 Dimensions dimension1 = {8, 1}; in TEST_F()
70 Dimensions dimension2 = {1, 8}; in TEST_F()
/third_party/boost/boost/units/
Dhomogeneous_system.hpp67 template<class System, class Dimensions>
72 template<class L, class Dimensions>
73 struct check_system<homogeneous_system<L>, Dimensions> :
78 Dimensions
86 template<class L, class Dimensions>
87 struct check_system<homogeneous_system<L>, Dimensions> : mpl::true_ {};
Dheterogeneous_system.hpp62 template<class L, class Dimensions, class Scale>
66 typedef Dimensions dimensions;
95 #define BOOST_UNITS_MAKE_HETEROGENEOUS_UNIT(BaseUnit, Dimensions) \ argument
97 Dimensions, \
107 Dimensions, \
240 template<class Dimensions, class System>
243 … typedef typename calculate_base_unit_exponents<typename System::type, Dimensions>::type exponents;
244 …e), the_specified_dimension_is_not_representible_in_the_given_system, (types<Dimensions, System>));
249 typedef heterogeneous_system<heterogeneous_system_impl<unit_list, Dimensions, no_scale> > type;
252 template<class Dimensions, class T>
[all …]
/third_party/boost/boost/geometry/core/
Dcoordinate_dimension.hpp93 template <typename Geometry, int Dimensions>
96 BOOST_STATIC_ASSERT(( static_cast<int>(dimension<Geometry>::value) == Dimensions )); in assert_dimension()
103 template <typename Geometry, int Dimensions>
106 BOOST_STATIC_ASSERT(( static_cast<int>(dimension<Geometry>::type::value) <= Dimensions )); in assert_dimension_less_equal()
109 template <typename Geometry, int Dimensions>
112 BOOST_STATIC_ASSERT(( static_cast<int>(dimension<Geometry>::type::value) >= Dimensions )); in assert_dimension_greater_equal()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Darithmetic_info.cc59 Dimensions sub_a_strategy = stra.at(0); in ExpendStrategy()
60 Dimensions sub_b_strategy = stra.at(1); in ExpendStrategy()
82 Dimensions sub_a_strategy = expend_strategy.at(0); in CheckStrategy()
83 Dimensions sub_b_strategy = expend_strategy.at(1); in CheckStrategy()
98 Dimensions sub_a_strategy = expend_strategy.at(0); in InferDevMatrixShape()
99 Dimensions sub_b_strategy = expend_strategy.at(1); in InferDevMatrixShape()
155 Dimensions sub_a_expend_strategy = expend_strategy.at(0); in InferTensorMap()
156 Dimensions sub_b_expend_strategy = expend_strategy.at(1); in InferTensorMap()
158 Dimensions sub_a_strategy = stra.at(0); in InferTensorMap()
159 Dimensions sub_b_strategy = stra.at(1); in InferTensorMap()
Dbias_add_info.cc35 Dimensions sub_a_strategy = stra.at(0); in CheckStrategy()
36 Dimensions sub_b_strategy = stra.at(1); in CheckStrategy()
48 Dimensions sub_a_strategy = stra.at(0); in InferDevMatrixShape()
62 Dimensions sub_a_strategy = stra.at(0); in InferTensorMap()
93 Dimensions input0_strategy = sp->GetInputDim()[0]; in GenerateOpStrategies()
96 Dimensions input1_strategy = {input0_strategy.at(1)}; in GenerateOpStrategies()
Dscatter_update_info.cc47 std::vector<Dimensions> stra = strategy->GetInputDim(); in CheckStrategy()
96 std::vector<Dimensions> stra = strategy_->GetInputDim(); in InferDevMatrixShape()
162 Dimensions first_input_strategy = sp->GetInputDim()[0]; in GenerateOpStrategies()
163 Dimensions indices_strategy(inputs_shape_[1].size(), 1); in GenerateOpStrategies()
165 Dimensions updates_strategy = indices_strategy; in GenerateOpStrategies()
Duniform_real_info.cc56 std::vector<Dimensions> stra = strategy->GetInputDim(); in CheckStrategy()
66 std::vector<Dimensions> stra = strategy_->GetInputDim(); in InferDevMatrixShape()
79 std::vector<Dimensions> stra = strategy_->GetInputDim(); in InferTensorMap()
128 std::vector<Dimensions> stra = strategy_->GetInputDim(); in UpdateShape()
145 std::vector<Dimensions> stra = strategy_->GetInputDim(); in ReplaceNodeInputOrAttrs()
Dmatmul_info.cc35 void SetDevMatrixShape(const Dimensions &mat_a_strategy, const Dimensions &mat_b_strategy, bool tra… in SetDevMatrixShape()
131 Status CheckRelevantDimension(const Dimensions &long_strategy, const Dimensions &short_strategy) { in CheckRelevantDimension()
159 Dimensions mat_a_strategy = stra.at(0); in CheckStrategy()
160 Dimensions mat_b_strategy = stra.at(1); in CheckStrategy()
208 Dimensions mat_a_strategy = stra.at(0); in InferDevMatrixShape()
209 Dimensions mat_b_strategy = stra.at(1); in InferDevMatrixShape()
427 Dimensions combined_partitions; in GenerateStrategies()
477 … mindspore::parallel::Dimensions combined_partitions, size_t input0_shape_size, in PrepareStrategy()
491 Dimensions input0_partitions, input1_partitions; in PrepareStrategy()
590 Dimensions batch_strategy(inputs_shape_[1].size() - 1, 1); in GenerateBatchStrategies()
Dgathernd_info.cc41 std::vector<Dimensions> stra = strategy->GetInputDim(); in CheckStrategy()
69 std::vector<Dimensions> stra = strategy_->GetInputDim(); in InferDevMatrixShape()
143 Dimensions indices_strategy = sp->GetInputDim()[0]; in GenerateOpStrategies()
144 Dimensions input_strategy(inputs_shape_[0].size(), 1); in GenerateOpStrategies()
Dgatherd_info.cc66 std::vector<Dimensions> stra = strategy->GetInputDim(); in CheckStrategy()
72 Dimensions input_strategy = stra[0]; in CheckStrategy()
73 Dimensions index_strategy = stra[1]; in CheckStrategy()
89 std::vector<Dimensions> stra = strategy_->GetInputDim(); in InferDevMatrixShape()
210 Dimensions first_input_strategy = sp->GetInputDim()[0]; in GenerateOpStrategies()
Dlayer_norm_info.cc70 Dimensions input_strategy = stra[LAYER_NORM_INPUT_INDEX]; in CheckStrategy()
71 Dimensions gamma_strategy = stra[LAYER_NORM_GAMMA_INDEX]; in CheckStrategy()
72 Dimensions beta_strategy = stra[LAYER_NORM_BETA_INDEX]; in CheckStrategy()
186 Dimensions input_strategy = sp->GetInputDim()[0]; in GenerateGammaAndBetaStrategies()
187 Dimensions gamma_strategy = input_strategy; in GenerateGammaAndBetaStrategies()
190 Dimensions beta_strategy = input_strategy; in GenerateGammaAndBetaStrategies()
Ddsd_matmul_info.cc94 Dimensions input_strategy = stra.at(0); in InferDevMatrixShape()
196 Dimensions input_w1_strategy = sp->GetInputDim()[0]; in GenerateOpStrategies()
197 Dimensions input_w2_strategy = input_w1_strategy; in GenerateOpStrategies()
198 Dimensions input_v_strategy = {input_w1_strategy[0], input_w1_strategy[1], 1, 1}; in GenerateOpStrategies()
Dresizebilinear_info.cc58 std::vector<Dimensions> stra = strategy->GetInputDim(); in CheckStrategy()
64 Dimensions input_strategy = stra[0]; in CheckStrategy()
82 std::vector<Dimensions> stra = strategy_->GetInputDim(); in InferDevMatrixShape()
Dreduce_method_info.cc36 Dimensions input_strategy = stra.at(0); in InferDevMatrixShape()
136 bool IsDataParallelStrategy(const Dimensions &strategy, int32_t stage_id) { in IsDataParallelStrategy()
147 Dimensions stra = strategy_->GetInputDim().at(0); in InferForwardCommunication()
227 Dimensions stra = strategy_->GetInputDim().at(0); in InferForwardCommunication()
330 Dimensions ReduceMethod::InferOutputStrategy() { in InferOutputStrategy()
332 Dimensions output_strategy; in InferOutputStrategy()
333 Dimensions stra = strategy_->GetInputDim().at(0); in InferOutputStrategy()
355 Dimensions output_strategy = InferOutputStrategy(); in InferTensorInfo()
484 Dimensions output_strategy = InferOutputStrategy(); in InferTensorInfo()
549 Dimensions stra = strategy->GetInputDim().at(0); in CheckStrategy()
Dtensordot_info.cc125 Dimensions input_a_strategy = stra[0]; in CheckStrategy()
126 Dimensions input_b_strategy = stra[1]; in CheckStrategy()
152 Dimensions input_a_strategy = stra.at(0); in InferDevMatrixShape()
153 Dimensions input_b_strategy = stra.at(1); in InferDevMatrixShape()
333 Dimensions input_a_strategy(inputs_shape_[0].size(), 1); in GenerateBatchStrategies()
334 Dimensions input_b_strategy(inputs_shape_[1].size(), 1); in GenerateBatchStrategies()
Dselect_info.cc48 std::vector<Dimensions> stra = strategy->GetInputDim(); in CheckStrategy()
64 std::vector<Dimensions> stra = strategy_->GetInputDim(); in InferDevMatrixShape()
119 Dimensions first_input_strategy = sp->GetInputDim()[0]; in GenerateOpStrategies()
/third_party/boost/boost/graph/
Dgrid_graph.hpp274 template < std::size_t Dimensions, typename VertexIndex = std::size_t,
280 typedef boost::array< bool, Dimensions > WrapDimensionArray;
284 typedef grid_graph< Dimensions, VertexIndex, EdgeIndex > type;
292 typedef boost::array< VertexIndex, Dimensions > vertex_descriptor;
378 inline std::size_t dimensions() const { return (Dimensions); } in dimensions()
453 for (std::size_t dimension_index = 0; dimension_index < Dimensions; in index_of()
469 boost::array< vertices_size_type, Dimensions > vertex; in vertex_at()
472 for (std::size_t dimension_index = 0; dimension_index < Dimensions; in vertex_at()
532 dimension_index_iter < Dimensions; ++dimension_index_iter) in edge_at()
643 for (std::size_t dimension_index = 0; dimension_index < Dimensions; in index_of()
[all …]
/third_party/skia/third_party/externals/abseil-cpp/absl/types/
Dvariant_benchmark.cc94 template <std::size_t NumIndices, class Dimensions>
108 template <std::size_t NumIndices, std::size_t... Dimensions>
109 struct MakeVariantTuple<NumIndices, absl::index_sequence<Dimensions...>> {
111 std::tuple<always_t<VariantOfAlternatives<NumIndices>, Dimensions>...>;
114 return std::make_tuple(MakeVariant<NumIndices>(Dimensions, index)...); in Run()
/third_party/abseil-cpp/absl/types/
Dvariant_benchmark.cc94 template <std::size_t NumIndices, class Dimensions>
108 template <std::size_t NumIndices, std::size_t... Dimensions>
109 struct MakeVariantTuple<NumIndices, absl::index_sequence<Dimensions...>> {
111 std::tuple<always_t<VariantOfAlternatives<NumIndices>, Dimensions>...>;
114 return std::make_tuple(MakeVariant<NumIndices>(Dimensions, index)...); in Run()

1234