Home
last modified time | relevance | path

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

/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::__anonae0b52e30111::GemvConfig
139 explicit GemvConfig(std::string name, PrimitiveType scalar_type, in GemvConfig()
226 explicit Config(PrimitiveType scalar_type, int64_t tile_rows, in Config()
454 explicit Config(PrimitiveType scalar_type, int64_t tile_rows, in Config()
665 explicit Config(PrimitiveType scalar_type, Dimensions dims, in Config()
684 PrimitiveType scalar_type() const { return scalar_type_; } in scalar_type() function in xla::cpu::__anonae0b52e30111::TiledSmallGemmEmitter::Config
762 PrimitiveType scalar_type() const { return config().scalar_type(); } in scalar_type() function in xla::cpu::__anonae0b52e30111::TiledSmallGemmEmitter
1002 void EmitRowMajorGemv(PrimitiveType scalar_type, int64_t tile_rows, in EmitRowMajorGemv()
1029 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.cc35 ShapedType scalar_type = RankedTensorType::get({}, element_type); in CreateConstOpWithSingleValue() local
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Ddata_type.cc27 std::string ToGlslType(const std::string& scalar_type, in ToGlslType()
/external/rust/crates/pdl-compiler/src/backends/rust_no_allocation/
Dpacket_parser.rs163 let scalar_type = get_integer_type(*width); in generate_packet() localVariable
/external/mesa3d/src/gallium/auxiliary/gallivm/
Dlp_bld_logic.c500 LLVMTypeRef scalar_type; in lp_build_any_true_range() local
Dlp_bld_nir_aos.c271 LLVMTypeRef scalar_type = LLVMInt8TypeInContext(gallivm->context); in emit_load_ubo() local
Dlp_bld_nir_soa.c509 LLVMTypeRef scalar_type = LLVMFloatTypeInContext(gallivm->context); in emit_load_var() local
/external/tensorflow/tensorflow/dtensor/mlir/
Dvalue_utils.cc76 auto scalar_type = in ReshapeSizeTypeToScalar() local
/external/mesa3d/src/intel/compiler/elk/
Delk_fs_bank_conflicts.cpp80 typedef int16_t scalar_type; typedef
188 typedef int16_t scalar_type; typedef
/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/ComputeLibrary/src/cpu/kernels/elementwise_binary/generic/neon/
Dimpl.h37 using scalar_type = typename VectorType::scalar_type; in elementwise_arithm_op() local
269 using scalar_type = typename VectorType::scalar_type; in elementwise_arithm_op() 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/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlower_tf.cc1614 auto scalar_type = in matchAndRewrite() local
/external/flatbuffers/src/
Didl_gen_cpp.cpp3588 auto scalar_type = GenUnderlyingCast(field, false, arg_name); in GenStructConstructor() local
/external/mesa3d/src/compiler/
Dglsl_types.c370 const glsl_type *scalar_type = glsl_get_base_glsl_type(type); in glsl_get_scalar_type() local
/external/mesa3d/src/compiler/spirv/
Dspirv_to_nir.c4302 const struct glsl_type *scalar_type = in vtn_composite_extract() local
/external/deqp/external/openglcts/modules/gl/
Dgl4cGPUShaderFP64Tests.cpp14039 const Utils::_variable_type scalar_type = Utils::getDoubleVariableType(1, 1); in getFunctionObject() local