| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | constant_utils.cc | 31 ShapedType scalar_type = RankedTensorType::get({}, element_type); in CreateConstOpWithSingleValue() local
37 attr = DenseElementsAttr::get(scalar_type, floatValues); in CreateConstOpWithSingleValue()
42 attr = DenseElementsAttr::get(scalar_type, floatValues); in CreateConstOpWithSingleValue()
45 DenseElementsAttr::get<float>(scalar_type, static_cast<float>(value)); in CreateConstOpWithSingleValue()
63 attr = mlir::OpaqueElementsAttr::get(dialect, scalar_type, mangled); in CreateConstOpWithSingleValue()
71 attr = DenseElementsAttr::get<int8_t>(scalar_type, in CreateConstOpWithSingleValue()
75 attr = DenseElementsAttr::get<int16_t>(scalar_type, in CreateConstOpWithSingleValue()
79 attr = DenseElementsAttr::get<int32_t>(scalar_type, in CreateConstOpWithSingleValue()
83 attr = DenseElementsAttr::get<int64_t>(scalar_type, in CreateConstOpWithSingleValue()
94 return rewriter->create<ConstantOp>(loc, scalar_type, attr); in CreateConstOpWithSingleValue()
|
| /external/OpenCL-CTS/test_conformance/subgroups/ |
| D | subhelpers.h | 399 using scalar_type = cl_int;
408 using scalar_type = cl_int;
414 using scalar_type = cl_int;
421 using scalar_type = cl_int;
428 using scalar_type = cl_int;
441 using scalar_type = cl_uint;
451 using scalar_type = cl_uint;
457 using scalar_type = cl_uint;
464 using scalar_type = cl_uint;
471 using scalar_type = cl_uint;
[all …]
|
| /external/mesa3d/src/intel/compiler/ |
| D | brw_fs_bank_conflicts.cpp | 80 typedef int16_t scalar_type; typedef
85 const scalar_type max_scalar = INT16_MAX;
90 const unsigned vector_width = 2 * sizeof(__m128i) / sizeof(scalar_type);
96 set(vector_type &v, unsigned i, scalar_type x) in set()
105 scalar_type
109 scalar_type x; in get()
156 scalar_type
188 typedef int16_t scalar_type; typedef
193 const scalar_type max_scalar = INT16_MAX;
204 set(vector_type &v, unsigned i, scalar_type x) in set()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | tiled_dot_emitter.cc | 112 PrimitiveType scalar_type() const { in scalar_type() function
113 return derived().config().scalar_type(); in scalar_type()
125 PrimitiveType scalar_type() const { return scalar_type_; } in scalar_type() function in xla::cpu::__anon47412fb60111::GemvConfig
133 return absl::StrCat(name_, "_", PrimitiveType_Name(scalar_type()), "_", in GetCacheKey()
139 explicit GemvConfig(string name, PrimitiveType scalar_type, int64 tile_rows, in GemvConfig() argument
142 scalar_type_(scalar_type), in GemvConfig()
225 explicit Config(PrimitiveType scalar_type, int64 tile_rows, int64 tile_cols, in Config() argument
227 : GemvConfig(/*name=*/"col_major_gemv", scalar_type, in Config()
243 vsl_(config.scalar_type(), /*vector_size=*/config.tile_rows(), b_, "") { in ColumnMajorMatrixVectorProductEmitter()
451 explicit Config(PrimitiveType scalar_type, int64 tile_rows, int64 tile_cols, in Config() argument
[all …]
|
| D | tiled_dot_emitter.h | 29 void EmitRowMajorGemv(PrimitiveType scalar_type, tensorflow::int64 tile_rows,
36 void EmitColumnMajorGemv(PrimitiveType scalar_type, tensorflow::int64 tile_rows,
43 void EmitSmallGemm(PrimitiveType scalar_type, tensorflow::int64 m,
|
| D | vector_support_library.cc | 51 if (type != scalar_type() && type != vector_type()) { in AssertCorrectTypes()
52 LOG(FATAL) << "Expected either " << TypeToString(scalar_type()) << " or " in AssertCorrectTypes()
148 is_vector ? vector_type() : scalar_type(), name()); in I1ToFloat()
152 CHECK(scalar_type()->isFloatingPointTy()); in IntegerTypeForFloatSize()
155 int64 float_size_bits = data_layout.getTypeSizeInBits(scalar_type()); in IntegerTypeForFloatSize()
165 CHECK_EQ(x->getType(), scalar_type()); in BroadcastScalar()
415 return llvm::Constant::getNullValue(scalar_type()); in GetZeroScalar()
|
| D | vector_support_library.h | 149 return BroadcastScalar(GetConstantFloat(scalar_type(), d)); in BroadcastScalar()
232 llvm::Type* scalar_type() const { return scalar_type_; } in scalar_type() function
321 : LlvmVariable(vector_support->scalar_type(), vector_support->b()) { in ScalarVariable()
|
| /external/eigen/unsupported/Eigen/src/FFT/ |
| D | ei_fftw_impl.h | 53 typedef float scalar_type;
70 void fwd(complex_type * dst,scalar_type * src,int nfft) {
75 void inv(scalar_type * dst,complex_type * src,int nfft) {
96 typedef double scalar_type;
113 void fwd(complex_type * dst,scalar_type * src,int nfft) {
118 void inv(scalar_type * dst,complex_type * src,int nfft) {
137 typedef long double scalar_type;
154 void fwd(complex_type * dst,scalar_type * src,int nfft) {
159 void inv(scalar_type * dst,complex_type * src,int nfft) {
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | unfuse_batch_norm.cc | 89 auto scalar_type = RankedTensorType::get({}, fp_type); in MaterializeEpsilon() local
91 DenseElementsAttr::get(scalar_type, {epsilon_attr.cast<Attribute>()}); in MaterializeEpsilon()
|
| /external/mesa3d/src/gallium/auxiliary/gallivm/ |
| D | lp_bld_logic.c | 496 LLVMTypeRef scalar_type; in lp_build_any_true_range() local
503 scalar_type = LLVMIntTypeInContext(bld->gallivm->context, in lp_build_any_true_range()
505 val = LLVMBuildBitCast(builder, val, scalar_type, ""); in lp_build_any_true_range()
|
| /external/vulkan-validation-layers/layers/ |
| D | shader_validation.cpp | 475 auto scalar_type = src->get_def(insn.word(2)); in GetLocationsConsumedByType() local
477 …(scalar_type.opcode() == spv::OpTypeInt || scalar_type.opcode() == spv::OpTypeFloat) ? scalar_type… in GetLocationsConsumedByType()
516 auto scalar_type = src->get_def(insn.word(2)); in GetComponentsConsumedByType() local
518 …(scalar_type.opcode() == spv::OpTypeInt || scalar_type.opcode() == spv::OpTypeFloat) ? scalar_type… in GetComponentsConsumedByType()
|
| /external/mesa3d/src/compiler/ |
| D | glsl_types.cpp | 416 const glsl_type *scalar_type = type->get_base_type(); in get_scalar_type() local
417 if (scalar_type == error_type) in get_scalar_type()
420 return scalar_type; in get_scalar_type()
|
| /external/deqp/external/openglcts/modules/gl/ |
| D | gl4cGPUShaderFP64Tests.cpp | 13936 const Utils::_variable_type scalar_type = Utils::getDoubleVariableType(1, 1); in getFunctionObject() local
13967 variable_type /* arg1_type */, scalar_type /* arg2_type */, in getFunctionObject()
13968 scalar_type /* arg3_type */); in getFunctionObject()
14288 scalar_type /* arg2_type */); in getFunctionObject()
14306 scalar_type /* arg2_type */); in getFunctionObject()
14332 scalar_type /* arg2_type */); in getFunctionObject()
14521 variable_type /* res_type */, scalar_type /* arg1_type */, in getFunctionObject()
14522 scalar_type /* arg2_type */, variable_type /* arg3_type */); in getFunctionObject()
14546 function, "step", Math::step, variable_type /* res_type */, scalar_type /* arg1_type */, in getFunctionObject()
|
| /external/mesa3d/src/compiler/spirv/ |
| D | spirv_to_nir.c | 3755 const struct glsl_type *scalar_type = in vtn_composite_extract() local
3757 struct vtn_ssa_value *ret = vtn_create_ssa_value(b, scalar_type); in vtn_composite_extract()
|