Home
last modified time | relevance | path

Searched defs:scalar_type (Results 1 – 14 of 14) sorted by relevance

/external/ComputeLibrary/src/core/NEON/kernels/convolution/depthwise/
Ddepthwise_pack_parameters.cpp29 template <> struct DType<1> { using scalar_type = uint8_t; }; typedef
30 template <> struct DType<2> { using scalar_type = uint16_t; }; typedef
31 template <> struct DType<4> { using scalar_type = uint32_t; }; typedef
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dtiled_dot_emitter.cc112 PrimitiveType scalar_type() const { in scalar_type() function
125 PrimitiveType scalar_type() const { return scalar_type_; } in scalar_type() function in xla::cpu::__anon305e0d7b0111::GemvConfig
139 explicit GemvConfig(string name, PrimitiveType scalar_type, int64_t tile_rows, in GemvConfig()
225 explicit Config(PrimitiveType scalar_type, int64_t tile_rows, in Config()
452 explicit Config(PrimitiveType scalar_type, int64_t tile_rows, in Config()
658 explicit Config(PrimitiveType scalar_type, Dimensions dims, in Config()
677 PrimitiveType scalar_type() const { return scalar_type_; } in scalar_type() function in xla::cpu::__anon305e0d7b0111::TiledSmallGemmEmitter::Config
753 PrimitiveType scalar_type() const { return config().scalar_type(); } in scalar_type() function in xla::cpu::__anon305e0d7b0111::TiledSmallGemmEmitter
992 void EmitRowMajorGemv(PrimitiveType scalar_type, int64_t tile_rows, in EmitRowMajorGemv()
1019 void EmitColumnMajorGemv(PrimitiveType scalar_type, int64_t tile_rows, in EmitColumnMajorGemv()
[all …]
Dvector_support_library.h234 llvm::Type* scalar_type() const { return scalar_type_; } in scalar_type() function
/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dconstant_utils.cc31 ShapedType scalar_type = RankedTensorType::get({}, element_type); in CreateConstOpWithSingleValue() local
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dunfuse_batch_norm.cc90 auto scalar_type = RankedTensorType::get({}, fp_type); in MaterializeEpsilon() local
/external/mesa3d/src/gallium/auxiliary/gallivm/
Dlp_bld_logic.c496 LLVMTypeRef scalar_type; in lp_build_any_true_range() local
/external/mesa3d/src/intel/compiler/
Dbrw_fs_bank_conflicts.cpp80 typedef int16_t scalar_type; typedef
188 typedef int16_t scalar_type; typedef
/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/
Dtf_cpurt_passes.cc346 auto scalar_type = mlir::MemRefType::get({}, memref_type.getElementType()); in MemrefToScalar() local
/external/vulkan-validation-layers/layers/
Dshader_validation.cpp475 auto scalar_type = src->get_def(insn.word(2)); in GetLocationsConsumedByType() local
516 auto scalar_type = src->get_def(insn.word(2)); in GetComponentsConsumedByType() local
/external/ComputeLibrary/src/core/NEON/kernels/
DNEElementwiseOperationKernel.cpp184 using scalar_type = typename VectorType::scalar_type; in elementwise_arithm_op() typedef
1071 using scalar_type = typename VectorType::scalar_type; in elementwise_arithm_op() typedef
/external/mesa3d/src/compiler/
Dglsl_types.cpp416 const glsl_type *scalar_type = type->get_base_type(); in get_scalar_type() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlower_tf.cc1622 auto scalar_type = in matchAndRewrite() local
/external/mesa3d/src/compiler/spirv/
Dspirv_to_nir.c3755 const struct glsl_type *scalar_type = in vtn_composite_extract() local
/external/deqp/external/openglcts/modules/gl/
Dgl4cGPUShaderFP64Tests.cpp13907 const Utils::_variable_type scalar_type = Utils::getDoubleVariableType(1, 1); in getFunctionObject() local