Home
last modified time | relevance | path

Searched defs:element_type (Results 1 – 25 of 107) sorted by relevance

12345

/third_party/mindspore/mindspore-src/source/mindspore/core/ir/dtype/
Dtensor_type.h63 void set_element(const TypePtr &element_type) { element_type_ = element_type; } in set_element()
104 void set_element(const TypePtr &element_type) { element_type_ = element_type; } in set_element()
134 explicit AnyType(const TypePtr &element_type) : TensorType(element_type) {} in AnyType()
162 explicit NegligibleType(const TypePtr &element_type) : AnyType(element_type) {} in NegligibleType()
194 virtual TypePtr element_type() { in element_type() function
246 void set_element(const TypePtr &element_type) { element_type_ = element_type; } in set_element()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/
Dkldiv_loss_info.cc127 auto element_type = outputs_dtype_->cast<mindspore::TensorTypePtr>()->element(); in InferForwardCommunication() local
133 auto element_type = outputs_dtype_->cast<mindspore::TensorTypePtr>()->element(); in InferForwardCommunication() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_frontend_func_impl/
Dunstack_ext.cc27 const ValuePtr &axis_value, const TypePtr &element_type) { in InferOutputForRankKnown()
84 auto element_type = tensor_type->element()->Clone(); in InferAbstract() local
/third_party/skia/third_party/externals/spirv-tools/source/fuzz/
Dfuzzer_pass_add_composite_inserts.cpp206 [](const opt::analysis::Type* element_type) { in ContainsPointer()
224 [](const opt::analysis::Type* element_type) { in ContainsRuntimeArray()
Dtransformation_add_type_array.cpp40 auto element_type = in IsApplicable() local
/third_party/skia/third_party/externals/swiftshader/third_party/SPIRV-Tools/source/fuzz/
Dfuzzer_pass_add_composite_inserts.cpp206 [](const opt::analysis::Type* element_type) { in ContainsPointer()
224 [](const opt::analysis::Type* element_type) { in ContainsRuntimeArray()
Dtransformation_add_type_array.cpp40 auto element_type = in IsApplicable() local
/third_party/spirv-tools/source/fuzz/
Dfuzzer_pass_add_composite_inserts.cpp207 [](const opt::analysis::Type* element_type) { in ContainsPointer()
225 [](const opt::analysis::Type* element_type) { in ContainsRuntimeArray()
Dtransformation_add_type_array.cpp40 auto element_type = in IsApplicable() local
/third_party/mesa3d/src/compiler/glsl/
Dgl_nir_link_uniform_initializers.c44 const struct glsl_type *element_type = glsl_get_array_element(type); in set_opaque_binding() local
205 const struct glsl_type *element_type = glsl_get_array_element(type); in set_uniform_initializer() local
221 const struct glsl_type *element_type = glsl_get_array_element(type); in set_uniform_initializer() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dmake_range.cc54 auto element_type = element->GetType(); in CheckMakeRangeInput() local
121 auto element_type = element->GetType(); in InferImplMakeRange() local
/third_party/spirv-tools/source/opt/
Dconstants.cpp397 const Type* element_type = type->AsVector()->element_type(); in GetNullCompositeConstant() local
404 const Type* element_type = type->AsMatrix()->element_type(); in GetNullCompositeConstant() local
414 const Type* element_type = type->AsArray()->element_type(); in GetNullCompositeConstant() local
432 const auto* element_type = type->element_type(); in GetNumericVectorConstantWithWords() local
539 const analysis::Type* element_type = vector_type->element_type(); in GetVectorComponents() local
Dtypes.h277 const Type* element_type() const { return element_type_; } in element_type() function
298 const Type* element_type() const { return element_type_; } in element_type() function
397 const Type* element_type() const { return element_type_; } in element_type() function
422 const Type* element_type() const { return element_type_; } in element_type() function
Dtype_manager.cpp978 const ForwardPointer* element_type = in ReplaceForwardPointers() local
985 const ForwardPointer* element_type = in ReplaceForwardPointers() local
1039 const Type* element_type = type->AsArray()->element_type(); in ReplaceType() local
1045 const Type* element_type = type->AsRuntimeArray()->element_type(); in ReplaceType() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/
Dkernel.cc373 auto element_type = tensor_type_ptr->element(); in SetType() local
382 TypePtr element_type = nullptr; in SetType() local
401 TypePtr element_type = nullptr; in SetType() local
423 void KernelTensor::SetSequenceDType(const TypePtr &element_type) { in SetSequenceDType()
/third_party/mindspore/mindspore-src/source/mindspore/core/mindapi/src/
Dtype.cc44 TensorType::TensorType(const TypePtr &element_type) in TensorType()
/third_party/gn/src/base/mac/
Dscoped_typeref.h54 typedef T element_type; typedef
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/
Dapply_momentum_scale_fusion.cc48 auto element_type = element->type_id(); in IsScalar() local
/third_party/skia/third_party/externals/spirv-tools/source/opt/
Dtypes.h271 const Type* element_type() const { return element_type_; } in element_type() function
293 const Type* element_type() const { return element_type_; } in element_type() function
395 const Type* element_type() const { return element_type_; } in element_type() function
420 const Type* element_type() const { return element_type_; } in element_type() function
Dtype_manager.cpp945 const ForwardPointer* element_type = in ReplaceForwardPointers() local
952 const ForwardPointer* element_type = in ReplaceForwardPointers() local
1006 const Type* element_type = type->AsArray()->element_type(); in ReplaceType() local
1012 const Type* element_type = type->AsRuntimeArray()->element_type(); in ReplaceType() local
/third_party/skia/third_party/externals/swiftshader/third_party/SPIRV-Tools/source/opt/
Dtypes.h271 const Type* element_type() const { return element_type_; } in element_type() function
293 const Type* element_type() const { return element_type_; } in element_type() function
395 const Type* element_type() const { return element_type_; } in element_type() function
420 const Type* element_type() const { return element_type_; } in element_type() function
Dtype_manager.cpp945 const ForwardPointer* element_type = in ReplaceForwardPointers() local
952 const ForwardPointer* element_type = in ReplaceForwardPointers() local
1006 const Type* element_type = type->AsArray()->element_type(); in ReplaceType() local
1012 const Type* element_type = type->AsRuntimeArray()->element_type(); in ReplaceType() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ir/
Ddtype_extends.cc249 auto element_type = StringToType(element_str); in TensorStrToType() local
268 auto element_type = StringToType(element_str); in RowTensorStrToType() local
358 auto element_type = StringToType(element_str); in UndeterminedStrToType() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/common/
Dfallback.cc113 auto element_type = type_elements[it]; in GenerateAbstractSequence() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/
Dgenerator.cc65 auto element_type = input_types->dynamic_element_type(); in CheckValidation() local

12345