Home
last modified time | relevance | path

Searched defs:to_type (Results 1 – 25 of 29) sorted by relevance

12

/external/pytorch/aten/src/ATen/
Dautocast_mode.h361 at::ScalarType to_type, in set_opt_dtype()
368 inline T set_opt_dtype(at::ScalarType to_type, T arg) { in set_opt_dtype()
383 at::ScalarType to_type, in type_from_firstarg()
Dautocast_mode.cpp120 Tensor cached_cast(at::ScalarType to_type, const Tensor& arg, DeviceType device_type) { in cached_cast()
DSparseCsrTensorUtils.h398 inline Tensor to_type(const Tensor& input, ScalarType dtype) { in to_type() function
/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/hardwares/
Dsimple_hardware.cc38 auto to_type = GetTypeId(); in GetHardwareSwitchingCost() local
Dcpu_hardware.cc59 auto to_type = GetTypeId(); in GetHardwareSwitchingCost() local
Dgpu_hardware.cc46 auto to_type = GetTypeId(); in GetHardwareSwitchingCost() local
/external/pytorch/torch/csrc/jit/passes/onnx/
Dcast_all_constant_to_floating.cpp33 int to_type = 0; in CastAllConstantToFloating() local
/external/tensorflow/tensorflow/compiler/xla/
Dprimitive_util.h209 PrimitiveType to_type) { in CastPreservesValues()
Dprimitive_util_test.cc291 auto to_type = static_cast<PrimitiveType>(to_type_int); in TEST() local
/external/pytorch/torch/testing/_internal/
Dautocast_test_lists.py396 def cast(val, to_type): argument
Dcommon_nn.py3723 def to_type(tensor, real, complex): function
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_element_type_converter.cc67 PrimitiveType to_type) { in GetConvertedTupleShape()
Delemental_ir_emitter.cc227 PrimitiveType to_type, llvm::Module* module, in EmitIntegralToFloating()
259 PrimitiveType to_type = op->shape().element_type(); in EmitIntegerUnaryOp() local
304 PrimitiveType to_type = op->shape().element_type(); in EmitIntegerUnaryOp() local
379 PrimitiveType to_type = op->shape().element_type(); in EmitFloatUnaryOp() local
469 PrimitiveType to_type = op->shape().element_type(); in EmitFloatUnaryOp() local
595 PrimitiveType to_type = op->shape().element_type(); in EmitComplexUnaryOp() local
/external/deqp/external/openglcts/modules/glesext/geometry_shader/
DesextcGeometryShaderLayeredFramebuffer.cpp523 glw::GLenum to_type = (n == 0) ? GL_UNSIGNED_BYTE : (n == 1) ? GL_INT : GL_UNSIGNED_INT; in iterate() local
589 glw::GLenum to_type = GL_NONE; in iterate() local
/external/libnl/include/nl-priv-dynamic-route/
Dnl-priv-dynamic-route.h47 enum rtnl_tc_type to_type; member
/external/tensorflow/tensorflow/core/kernels/mlir_generated/
Dgpu_unary_ops_test.cc191 #define TEST_CAST_FROM_TO(from_type, to_type) \ argument
202 #define TEST_NON_NEGATIVE_VALUES_CAST_FROM_TO(from_type, to_type) \ argument
/external/mesa3d/src/compiler/glsl/
Dir_function.cpp161 const glsl_type *to_type; in get_parameter_match_type() local
Dgl_nir_link_functions.c128 const struct glsl_type *to_type; in get_parameter_match_type() local
/external/pytorch/test/
Dtest_xpu.py332 def cast(val, to_type): argument
/external/pytorch/torch/onnx/
Dsymbolic_opset8.py185 def _cast_to_type(g: jit_utils.GraphContext, input, to_type): argument
/external/coreboot/src/lib/
Dubsan.c253 struct ubsan_type_descriptor *to_type; member
/external/openthread/third_party/mbedtls/repo/library/
Dpk.c684 psa_key_type_t to_type = psa_get_key_type(attributes); in import_pair_into_psa() local
/external/libchrome/base/debug/
Dactivity_tracker.cc78 uint32_t to_type) { in AllocateFrom()
/external/flatbuffers/src/
Didl_gen_kotlin.cpp789 std::string to_type; in FlexBufferBuilderCast() local
/external/python/pycparser/pycparser/
Dc_ast.py292 def __init__(self, to_type, expr, coord=None): argument

12