Home
last modified time | relevance | path

Searched defs:TENSOR (Results 1 – 25 of 26) sorted by relevance

12

/external/pytorch/aten/src/ATen/native/transformers/hip/
Daotriton_adapter.h12 #define CHECK_NOSPARSE_CONTIGUOUS_CUDA(TENSOR) \ argument
17 #define CHECK_NOSPARSE_LASTCONTIGUOUS_CUDA(TENSOR) \ argument
/external/pytorch/aten/src/ATen/
DUtils.cpp58 #define TENSOR(T, _1) \ macro
69 #define TENSOR(T, _1) \ macro
/external/pytorch/torch/csrc/api/include/torch/detail/
DTensorDataContainer.h122 #define TENSOR(T, S) \ macro
165 #define TENSOR(T, S) \ macro
194 #define TENSOR(T, S) \ in AT_FORALL_SCALAR_TYPES_AND3() macro
/external/pytorch/aten/src/ATen/native/transformers/cuda/mem_eff_attention/
Dgemm_kernel_utils.h62 #define CHECK_NOSPARSE_CONTIGUOUS_CUDA(TENSOR) \ argument
67 #define CHECK_NOSPARSE_LASTCONTIGUOUS_CUDA(TENSOR) \ argument
/external/pytorch/aten/src/ATen/ops/
Dtensor.h8 #define TENSOR(T, S) \ macro
/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
Dkernel_fallback_execute.h36 TENSOR = 0, // Output type is tensorflow::Tensor enumerator
/external/google-cloud-java/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/proto/google/cloud/aiplatform/v1beta1/
Dtensorboard_time_series.proto66 TENSOR = 2; enumerator
/external/googleapis/google/cloud/aiplatform/v1beta1/
Dtensorboard_time_series.proto66 TENSOR = 2; enumerator
/external/google-cloud-java/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/proto/google/cloud/aiplatform/v1/
Dtensorboard_time_series.proto66 TENSOR = 2; enumerator
/external/googleapis/google/cloud/aiplatform/v1/
Dtensorboard_time_series.proto66 TENSOR = 2; enumerator
/external/executorch/backends/vulkan/runtime/graph/containers/
DTypes.h29 TENSOR, enumerator
/external/pytorch/torch/distributed/checkpoint/
Dplanner.py34 TENSOR = auto() variable in WriteItemType
40 TENSOR = auto() variable in LoadItemType
/external/tensorflow/tensorflow/python/framework/
Dop_def_util.h40 TENSOR, // "tensor" (tf.TensorProto) enumerator
/external/executorch/devtools/etdump/
Dschema_flatcc.py77 TENSOR = "Tensor" variable in ValueType
/external/pytorch/torch/csrc/inductor/aoti_eager/
Dkernel_meta_info.h86 TENSOR, enumerator
/external/ComputeLibrary/src/core/CL/cl_kernels/
Dtile_helpers.h489 #define V_LOAD(DATA_TYPE, WIDTH, TENSOR_TYPE, TENSOR, X, Y, STRIDE_Y) V_LOAD_STR(DATA_TYPE, WIDTH, … argument
490 #define V_LOAD_STR(DATA_TYPE, WIDTH, TENSOR_TYPE, TENSOR, X, Y, STRIDE_Y) V_LOAD_##TENSOR_TYPE(DATA… argument
491 #define V_LOAD_BUFFER(DATA_TYPE, WIDTH, TENSOR, X, Y, STRIDE_Y) \ argument
494 #define V_LOAD_IMAGE(DATA_TYPE, WIDTH, TENSOR, X, Y, STRIDE_Y) READ_IMAGE2D(DATA_TYPE, CONVERT_VECT… argument
508 #define V_STORE(DATA_TYPE, WIDTH, TENSOR_TYPE, TENSOR, X, Y, STRIDE_Y, VALUES) V_STORE_STR(DATA_TYP… argument
509 #define V_STORE_STR(DATA_TYPE, WIDTH, TENSOR_TYPE, TENSOR, X, Y, STRIDE_Y, VALUES) V_STORE_##TENSOR… argument
510 #define V_STORE_BUFFER(DATA_TYPE, WIDTH, TENSOR, X, Y, STRIDE_Y, VALUES) \ argument
513 #define V_STORE_IMAGE(DATA_TYPE, WIDTH, TENSOR, X, Y, STRIDE_Y, VALUES) WRITE_IMAGE2D(DATA_TYPE, CO… argument
531 #define T_LOAD(DATA_TYPE, HEIGHT, WIDTH, TENSOR_TYPE, TENSOR, X, Y, YI_MULTIPLIER, STRIDE_Y, dst) … argument
552 #define T_LOAD_INDIRECT(DATA_TYPE, HEIGHT, WIDTH, TENSOR_TYPE, TENSOR, X, STRIDE_Y, indirect_y, dst… argument
[all …]
/external/executorch/codegen/tools/
Dgen_oplist.py76 TENSOR = 5 variable in KernelType
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dweights.h170 TENSOR = 0, enumerator
/external/google-cloud-java/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/java/com/google/cloud/aiplatform/v1/
DTimeSeriesDataPoint.java76 TENSOR(4), enumConstant
DTensorboardTimeSeries.java117 TENSOR(2), enumConstant
/external/google-cloud-java/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/java/com/google/cloud/aiplatform/v1beta1/
DTimeSeriesDataPoint.java76 TENSOR(4), enumConstant
DTensorboardTimeSeries.java117 TENSOR(2), enumConstant
/external/pytorch/torch/csrc/utils/
Dpython_arg_parser.h100 TENSOR, enumerator
/external/tensorflow/tensorflow/compiler/mlir/tfr/python/
Dtfr_gen.py58 TENSOR = 1 variable in TFRTypes
/external/libaom/av1/encoder/
Dcnn.c43 } TENSOR; typedef

12