Home
last modified time | relevance | path

Searched defs:dims (Results 1 – 25 of 197) sorted by relevance

12345678

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/
Dutils.h25 static inline size_t GetInputOffset(int num_dims, const int *dims, const int *iter) { in GetInputOffset()
34 static inline size_t GetOutputOffset(int num_dims, const int *dims, const int *iter, int num_axis, … in GetOutputOffset()
53 static inline int NextIndex(int num_dims, const int *dims, int *current) { in NextIndex()
/third_party/mindspore/mindspore/core/ops/fusion/
Dtile_fusion.cc23 void TileFusion::Init(const std::vector<int64_t> &dims) { this->set_dims(dims); } in Init()
25 void TileFusion::set_dims(const std::vector<int64_t> &dims) { (void)this->AddAttr(kDims, MakeValue(… in set_dims()
/third_party/mindspore/mindspore/lite/mindir/src/
Dmindir_tensor.cc29 std::vector<int32_t> dims(1, 0); in MindIR_Tensor_Create() local
41 …MindIR_Tensor_Create(const std::string &name, DataType data_type, const std::vector<int32_t> &dims, in MindIR_Tensor_Create()
74 flatbuffers::Offset<flatbuffers::Vector<int32_t>> dims; in MindIR_Tensor_SetName() local
116 flatbuffers::Offset<flatbuffers::Vector<int32_t>> dims; in MindIR_Tensor_SetDataType() local
165 void MindIR_Tensor_SetDims(TensorPtr *tensor, const std::vector<int32_t> &dims) { in MindIR_Tensor_SetDims()
212 flatbuffers::Offset<flatbuffers::Vector<int32_t>> dims; in MindIR_Tensor_SetFormat() local
299 flatbuffers::Offset<flatbuffers::Vector<int32_t>> dims; in MindIR_Tensor_SetData() local
351 flatbuffers::Offset<flatbuffers::Vector<int32_t>> dims; in MindIR_Tensor_SetQuantParams() local
/third_party/mindspore/mindspore/lite/src/delegate/tensorrt/op/
Dconvolution_tensorrt.cc131 auto dims = ConvertCudaDims(stride_val); in SetAttributes() local
138 auto dims = ConvertCudaDims(dilation_val); in SetAttributes() local
153 nvinfer1::Dims dims{}; in SetAttributes() local
/third_party/mindspore/mindspore/lite/src/delegate/tensorrt/
Dtensorrt_utils.cc23 nvinfer1::Dims dims{}; in ConvertCudaDims() local
36 nvinfer1::Dims dims{}; in ConvertCudaDims() local
49 bool SameDims(nvinfer1::Dims dims, const std::vector<int64_t> &shape) { in SameDims()
65 std::vector<int64_t> ConvertMSShape(const nvinfer1::Dims dims) { in ConvertMSShape()
117 nvinfer1::Dims dims = ConvertCudaDims(ms_tensor.Shape()); in ConvertConstantTensor() local
136 nvinfer1::Dims dims = ConvertCudaDims(1, shape_size); in ConvertScalarToITensor() local
186 nvinfer1::Dims dims = ConvertCudaDims(shape); in ConvertTensorWithExpandDims() local
377 nvinfer1::Dims dims = trt_tensor->getDimensions(); in GetTensorFormat() local
/third_party/mindspore/mindspore/ccsrc/cxx_api/model/ms/
Dms_model.cc32 static std::string GenerateShapeKey(const std::vector<std::vector<int64_t>> &dims) { in GenerateShapeKey()
49 std::shared_ptr<GraphCell> MsModel::GenerateGraphCell(const std::vector<std::vector<int64_t>> &dims in GenerateGraphCell()
117 … MsModel::Resize(const std::vector<MSTensor> &inputs, const std::vector<std::vector<int64_t>> &dim… in Resize()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/
Dcommon_func.c27 int Offset4d(const int *shape, const int *dims) { return Offset(shape, dims[0], dims[1], dims[2], d… in Offset4d()
29 int Offset6d(const int *shape, const int *dims) { in Offset6d()
/third_party/boost/libs/mpi/test/
Dcartesian_topology_init_test.cpp43 int dims[] = {2,3,4}; in BOOST_AUTO_TEST_CASE() local
64 int dims[] = {2,3,4}; in BOOST_AUTO_TEST_CASE() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dl2_normalize_cpu_kernel.cc38 int dims = SizeToInt(input_shape_.size()); in InitKernel() local
51 …rmalizeCPUKernel<T>::CalcDenominator(const T *input_addr, const size_t reduce_size, const int dims, in CalcDenominator()
135 int dims = SizeToInt(input_shape_.size()); in Launch() local
/third_party/boost/boost/mpi/
Dcartesian_communicator.hpp138 cartesian_topology(std::vector<cartesian_dimension> const& dims) in cartesian_topology()
146 explicit cartesian_topology(InitArr dims) in cartesian_topology()
157 explicit cartesian_topology(std::initializer_list<cartesian_dimension> dims) in cartesian_topology()
/third_party/mesa3d/src/mesa/main/
Dtexstorage.c52 legal_texobj_target(const struct gl_context *ctx, GLuint dims, GLenum target) in legal_texobj_target()
308 GLuint dims, GLenum target, in tex_storage_error_check()
392 texture_storage(struct gl_context *ctx, GLuint dims, in texture_storage()
496 texture_storage_error(struct gl_context *ctx, GLuint dims, in texture_storage_error()
508 texture_storage_no_error(struct gl_context *ctx, GLuint dims, in texture_storage_no_error()
523 texstorage_error(GLuint dims, GLenum target, GLsizei levels, in texstorage_error()
564 texstorage_no_error(GLuint dims, GLenum target, GLsizei levels, in texstorage_no_error()
580 texturestorage_error(GLuint dims, GLuint texture, GLsizei levels, in texturestorage_error()
621 texturestorage_no_error(GLuint dims, GLuint texture, GLsizei levels, in texturestorage_no_error()
793 _mesa_texture_storage_memory(struct gl_context *ctx, GLuint dims, in _mesa_texture_storage_memory()
/third_party/boost/libs/mpi/src/
Dcartesian_communicator.cpp48 std::vector<int> dims(topology.size()); in cartesian_communicator() local
148 cartesian_topology::split(std::vector<int>& dims, std::vector<bool>& periodics) const { in split()
160 cartesian_dimensions(int sz, std::vector<int>& dims) { in cartesian_dimensions()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dparallel_fusion.h40 …ParallelInfo(const AnfNodePtrList &nodes, const std::vector<DimInfoPtr> &dims, const FusionInfoPtr… in ParallelInfo()
56 const std::vector<DimInfoPtr> &dims() const { return dims_; } in dims() function
/third_party/mesa3d/src/intel/compiler/
Dbrw_vec4_surface_builder.cpp146 unsigned dims, unsigned size, in emit_untyped_read()
163 unsigned dims, unsigned size, in emit_untyped_write()
184 unsigned dims, unsigned rsize, unsigned op, in emit_untyped_atomic()
/third_party/mesa3d/src/mesa/drivers/dri/nouveau/
Dnouveau_texture.c419 nouveau_compressed_copy(struct gl_context *ctx, GLint dims, in nouveau_compressed_copy()
444 nouveau_teximage(struct gl_context *ctx, GLint dims, in nouveau_teximage()
517 nouveau_teximage_123d(struct gl_context *ctx, GLuint dims, in nouveau_teximage_123d()
527 nouveau_compressed_teximage(struct gl_context *ctx, GLuint dims, in nouveau_compressed_teximage()
545 nouveau_texsubimage(struct gl_context *ctx, GLint dims, in nouveau_texsubimage()
597 nouveau_texsubimage_123d(struct gl_context *ctx, GLuint dims, in nouveau_texsubimage_123d()
610 nouveau_compressed_texsubimage(struct gl_context *ctx, GLuint dims, in nouveau_compressed_texsubimage()
/third_party/mindspore/mindspore/lite/tools/converter/quantizer/
Dmixed_bit_weight_quantizer.cc24 float MixedBitWeightQuantizer::MeasureQuantizationError(float *weights, const int *shape, int dims,… in MeasureQuantizationError()
96 …ult MixedBitWeightQuantizer::BinarySearchForQuantizationScale(float *weights, int *shape, int dims, in BinarySearchForQuantizationScale()
144 int dims = shape.size(); in DoQuantization() local
/third_party/mesa3d/src/mesa/swrast/
Ds_texfetch.c311 struct swrast_texture_image *texImage, GLuint dims) in set_fetch_functions()
350 GLuint dims; in _mesa_update_fetch_functions() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/
Dsplit_gpu_kernel.h65 int dims = SizeToInt(input_shape.size()); in Init() local
141 int dims = SizeToInt(input_shape.size()); in CheckParam() local
/third_party/mindspore/mindspore/ccsrc/utils/
Dcheckpoint.proto30 repeated int64 dims = 1; field
/third_party/mindspore/mindspore/lite/src/
Dexecutor.h42 const std::vector<std::vector<int>> &dims) { in Resize()
/third_party/mindspore/mindspore/lite/src/ops/populate/
Dtile_populate.cc39 auto dims = value->dims(); in PopulateTileParameter() local
/third_party/boost/boost/gil/
Dimage.hpp239 void recreate(const point_t& dims, std::size_t alignment = 0) in recreate()
264 void recreate(const point_t& dims, const Pixel& p_in, std::size_t alignment = 0) in recreate()
290 void recreate(const point_t& dims, std::size_t alignment, const Alloc alloc_in) in recreate()
315 … void recreate(const point_t& dims, const Pixel& p_in, std::size_t alignment, const Alloc alloc_in) in recreate()
473 void create_view(point_t const& dims, std::true_type) // is planar in create_view()
496 void create_view(point_t const& dims, std::false_type) // is planar in create_view()
/third_party/mindspore/mindspore/lite/tools/converter/parser/caffe/
Dcaffe_tile_parser.cc30 std::vector<int64_t> dims; in Parse() local
/third_party/mindspore/mindspore/lite/src/ops/populate/v0/
Dtile_populate_v0.cc40 auto dims = tile_prim->dims(); in PopulateTileParameter() local
/third_party/mindspore/mindspore/lite/tools/benchmark/
Dbenchmark_unified_api.h81 std::vector<int64_t> dims; in ConverterToInt64Vector() local

12345678