Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/runtime_fallback/util/
Dtype_util.h34 #define DTYPE(TFRT_ENUM, DT_ENUM) \ in GetTfDataType() macro
49 #define DTYPE(TFRT_ENUM, DT_ENUM) \ in GetTfrtDtype() macro
/external/executorch/kernels/test/
Dop_fill_test.cpp82 #define TEST_FILL_OUT(FN, DTYPE) \ argument
90 #define GENERATE_SCALAR_INPUT_SUPPORT_TEST(_, DTYPE) \ argument
98 #define GENERATE_TENSOR_INPUT_SUPPORT_TEST(_, DTYPE) \ in ET_FORALL_REAL_TYPES_AND() argument
Dop_scatter_add_test.cpp283 #define TEST_ENTRY(CTYPE, DTYPE) test_scatter_add_out<ScalarType::DTYPE>(); in TEST_F() argument
320 #define TEST_ENTRY(CTYPE, DTYPE) \ in TEST_F() argument
330 #define TEST_ENTRY(CTYPE, DTYPE) \ in TEST_F() argument
Dop_ones_test.cpp45 #define GENERATE_TEST(_, DTYPE) \ argument
Dop_scatter_test.cpp370 #define TEST_ENTRY(CTYPE, DTYPE) test_scatter_src_out<ScalarType::DTYPE>(); in TEST_F() argument
376 #define TEST_ENTRY(CTYPE, DTYPE) \ in TEST_F() argument
383 #define TEST_ENTRY(CTYPE, DTYPE) test_scatter_value_out<ScalarType::DTYPE>(); in TEST_F() argument
417 #define TEST_ENTRY(CTYPE, DTYPE) \ in TEST_F() argument
Dop_zeros_test.cpp44 #define GENERATE_TEST(_, DTYPE) \ argument
Dop_full_test.cpp64 #define GENERATE_TEST(_, DTYPE) \ argument
Dop_empty_test.cpp49 #define GENERATE_TEST(_, DTYPE) \ argument
Dop_tril_test.cpp717 #define GENERATE_GENERIC_TEST(_, DTYPE) \ argument
733 #define GENERATE_REAL_TEST(_, DTYPE) \ in ET_FORALL_REAL_TYPES_AND() argument
Dop_gather_test.cpp196 #define TEST_ENTRY(CTYPE, DTYPE) test_gather_out<ScalarType::DTYPE>(); in TEST_F() argument
231 #define TEST_ENTRY(CTYPE, DTYPE) \ in TEST_F() argument
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dconst_op.cc50 #define HANDLE_SPLAT(DTYPE, field_name, xla_type) \ in GetScalarConst() argument
79 #define HANDLE_COMPLEX_SPLAT(DTYPE, field_name, xla_type) \ in GetScalarConst() argument
/external/fec/
Dgen_ccsds_tal.c16 #define DTYPE unsigned char macro
/external/tensorflow/tensorflow/core/tfrt/utils/
Dtensor_util.cc89 #define DTYPE(TFRT_DTYPE, TF_DTYPE) \ in ConvertTFDTypeToTFRTDType() macro
101 #define DTYPE(TFRT_DTYPE, TF_DTYPE) \ in ConvertTFRTDTypeToTFDType() macro
Dutils.cc58 #define DTYPE(TFRT_DTYPE, TF_DTYPE) \ in ConvertTfDTypeToTfrtDType() macro
/external/tensorflow/tensorflow/python/framework/
Dop_def_util.h38 DTYPE, // "type" (tf.dtypes.DType) enumerator
/external/federated-compute/fcp/demo/
Dcheckpoint_tensor_reference_test.py27 DTYPE = tf.int32 variable
/external/federated-compute/fcp/aggregation/core/
Ddatatype.h72 #define MATCH_TYPE_AND_DTYPE(TYPE, DTYPE) \ argument
/external/libyuv/source/
Drow_any.cc192 #define ANY41PT(NAMEANY, ANY_SIMD, STYPE, SBPP, DTYPE, BPP, MASK) \ argument
501 #define ANY31PT(NAMEANY, ANY_SIMD, STYPE, SBPP, DTYPE, BPP, MASK) \ argument
1530 #define ANY11T(NAMEANY, ANY_SIMD, SBPP, BPP, STYPE, DTYPE, MASK) \ argument
1596 #define ANY11C(NAMEANY, ANY_SIMD, SBPP, BPP, STYPE, DTYPE, MASK) \ argument
/external/tensorflow/tensorflow/core/kernels/
Dspacetobatch_benchmark_test.cc61 #define BM_SpaceToBatchDev(OP, DEVICE, DTYPE, B, H, W, D, BS, P00, P01, P10, … argument
/external/tensorflow/tensorflow/core/transforms/utils/
Dop_cat_helper.cc30 #define IF_SPLAT_VALUE_IS(DTYPE, VALUE) \ in SplatElementsAttrHasValue() argument
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dconvert_tensor.cc126 #define CONVERT_FLAT(DTYPE, CTYPE) \ in ConvertTensor() argument
172 #define MATCH(DTYPE, FIELD) \ in NumberOfMaterializedElements() argument
/external/tensorflow/tensorflow/core/ir/importexport/
Dconvert_tensor.cc123 #define CONVERT_FLAT(DTYPE, CTYPE) \ in ConvertTensor() argument
170 #define MATCH(DTYPE, FIELD) \ in NumberOfMaterializedElements() argument
/external/tensorflow/tensorflow/core/grappler/
Dutils.cc409 #define HANDLE_DOUBLE_CASE(DTYPE) \ argument
418 #define HANDLE_INT_CASE(DTYPE) \ argument
/external/mesa3d/src/imagination/pco/
Dpco_ops.py36 DTYPE = enum_type('dtype', [ variable
/external/libaom/third_party/libyuv/source/
Drow_any.cc1013 #define ANY11C(NAMEANY, ANY_SIMD, SBPP, BPP, STYPE, DTYPE, MASK) \ argument

12