Home
last modified time | relevance | path

Searched defs:itensor (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/eigen/unsupported/test/
H A Dcxx11_tensor_casts.cpp41 Tensor<int, 2> itensor(20,30); in test_vectorized_cast() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/
H A DConv.cpp120 #define MKLDNNTensor(itensor, options) \ argument
/aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/
H A DNestedTensorMath.cpp717 for (int64_t itensor = 0; itensor < ntensors; itensor++) { in NestedTensor_compute_size_stride() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dconvert_nodes_test.cc251 ITensorProxyPtr itensor(dims); in TEST() local