Home
last modified time | relevance | path

Searched defs:scalar_type (Results 1 – 15 of 15) 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::__anon8d179a540111::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::__anon8d179a540111::TiledSmallGemmEmitter::Config
762 PrimitiveType scalar_type() const { return config().scalar_type(); } in scalar_type() function in xla::cpu::__anon8d179a540111::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/tensorflow/tensorflow/dtensor/mlir/
Dvalue_utils.cc76 auto scalar_type = in ReshapeSizeTypeToScalar() 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/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/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.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/spirv/
Dspirv_to_nir.c3755 const struct glsl_type *scalar_type = in vtn_composite_extract() local
/external/deqp/external/openglcts/modules/gl/
Dgl4cGPUShaderFP64Tests.cpp13904 const Utils::_variable_type scalar_type = Utils::getDoubleVariableType(1, 1); in getFunctionObject() local