Home
last modified time | relevance | path

Searched defs:scalar_type (Results 1 – 10 of 10) 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::__anon47412fb60111::GemvConfig
139 explicit GemvConfig(string name, PrimitiveType scalar_type, int64 tile_rows, in GemvConfig()
225 explicit Config(PrimitiveType scalar_type, int64 tile_rows, int64 tile_cols, in Config()
451 explicit Config(PrimitiveType scalar_type, int64 tile_rows, int64 tile_cols, in Config()
656 explicit Config(PrimitiveType scalar_type, Dimensions dims, in Config()
675 PrimitiveType scalar_type() const { return scalar_type_; } in scalar_type() function in xla::cpu::__anon47412fb60111::TiledSmallGemmEmitter::Config
751 PrimitiveType scalar_type() const { return config().scalar_type(); } in scalar_type() function in xla::cpu::__anon47412fb60111::TiledSmallGemmEmitter
990 void EmitRowMajorGemv(PrimitiveType scalar_type, int64 tile_rows, in EmitRowMajorGemv()
1017 void EmitColumnMajorGemv(PrimitiveType scalar_type, int64 tile_rows, in EmitColumnMajorGemv()
[all …]
Dvector_support_library.h232 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.cc89 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/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/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.cpp13936 const Utils::_variable_type scalar_type = Utils::getDoubleVariableType(1, 1); in getFunctionObject() local