Home
last modified time | relevance | path

Searched defs:ndim (Results 1 – 25 of 28) sorted by relevance

12

/external/tensorflow/tensorflow/python/framework/
Dfast_tensor_util.pyx10 tensor_proto, np.ndarray[np.uint16_t, ndim=1] nparray): argument
22 tensor_proto, np.ndarray[np.uint16_t, ndim=1] nparray): argument
30 tensor_proto, np.ndarray[np.float32_t, ndim=1] nparray): argument
38 tensor_proto, np.ndarray[np.float64_t, ndim=1] nparray): argument
46 tensor_proto, np.ndarray[np.int32_t, ndim=1] nparray): argument
53 tensor_proto, np.ndarray[np.uint32_t, ndim=1] nparray): argument
60 tensor_proto, np.ndarray[np.int64_t, ndim=1] nparray): argument
67 tensor_proto, np.ndarray[np.uint64_t, ndim=1] nparray): argument
74 tensor_proto, np.ndarray[np.uint8_t, ndim=1] nparray): argument
82 tensor_proto, np.ndarray[np.uint16_t, ndim=1] nparray): argument
[all …]
/external/python/pybind11/include/pybind11/
Dbuffer_info.h20 auto ndim = shape.size(); in PYBIND11_NAMESPACE_BEGIN() local
30 auto ndim = shape.size(); in f_strides() local
45 ssize_t ndim = 0; // Number of dimensions member
Dpytypes.h1514 size_t ndim = shape->size(); in from_buffer() local
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dtranspose.cc540 int ndim = a_dims.size(); in RemoveTrivialDimensions() local
596 int ndim = a_dims.size(); in CoalesceDimensions() local
669 int ndim, absl::Span<int64_t const> tiling_spec, in ParseTilingSpecification()
693 const int ndim = a_dims_.size(); in BuildPlanNodes() local
904 int ndim = dims.size(); in Create() local
1041 int ndim = a_dims_.size(); in Initialize() local
1217 int ndim = permutation_.size(); in ChooseParallelizationStrategy() local
/external/tensorflow/tensorflow/c/eager/
Ddlpack.cc250 int ndim) { in IsValidStrideCompactRowMajorData()
297 int ndim = tensor->dims(); in TFE_HandleToDLPack() local
/external/python/cpython3/Lib/test/
Dtest_buffer.py264 def strides_from_shape(ndim, shape, itemsize, layout): argument
321 def getindex(ndim, ind, strides): argument
435 def verify_structure(memlen, itemsize, ndim, shape, strides, offset): argument
564 def randslice_from_shape(ndim, shape): argument
698 def genslices_ndim(ndim, shape): argument
716 def rslices_ndim(ndim, shape, iterations=5): argument
774 ndim, shape, strides, argument
/external/tensorflow/tensorflow/python/keras/engine/
Dinput_spec.py74 ndim=None, argument
/external/tensorflow/tensorflow/python/layers/
Dutils.py22 def convert_data_format(data_format, ndim): argument
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcusolver_rewriter.cc58 int ndim = a_shape.dimensions_size(); in CreateCholesky() local
/external/tensorflow/tensorflow/compiler/xla/python/
Dpy_buffer.h133 int ndim() const { return buffer()->on_device_shape().dimensions_size(); } in ndim() function
/external/python/cpython3/Objects/
Dmemoryobject.c362 copy_rec(const Py_ssize_t *shape, Py_ssize_t ndim, Py_ssize_t itemsize, in copy_rec()
630 memory_alloc(int ndim) in memory_alloc()
701 int ndim) in mbuf_add_incomplete_view()
1259 copy_shape(Py_ssize_t *shape, const PyObject *seq, Py_ssize_t ndim, in copy_shape()
1298 cast_to_ND(PyMemoryViewObject *mv, const PyObject *shape, int ndim) in cast_to_ND()
1373 Py_ssize_t ndim = 1; in memoryview_cast_impl() local
2086 tolist_rec(const char *ptr, Py_ssize_t ndim, const Py_ssize_t *shape, in tolist_rec()
2793 Py_ssize_t ndim, const Py_ssize_t *shape, in cmp_rec()
/external/python/pybind11/tests/
Dtest_numpy_vectorize.cpp86 py::ssize_t ndim; in TEST_SUBMODULE() local
/external/tensorflow/tensorflow/python/keras/utils/
Dconv_utils.py27 def convert_data_format(data_format, ndim): argument
/external/python/cpython3/Modules/
D_testbuffer.c473 copy_rec(const Py_ssize_t *shape, Py_ssize_t ndim, Py_ssize_t itemsize, in copy_rec()
612 const Py_ssize_t *suboffsets, Py_ssize_t ndim, Py_ssize_t itemsize) in unpack_rec()
931 Py_ssize_t ndim) in verify_structure()
1121 Py_ssize_t ndim) in init_structure()
1181 Py_ssize_t ndim; in init_ndbuf() local
2483 Py_ssize_t ndim) in arraycmp()
/external/tensorflow/tensorflow/compiler/xla/service/
Deigh_expander.cc198 int ndim = shape.dimensions_size(); in PermuteRowsInColumn() local
218 int ndim = shape.dimensions_size(); in PermuteColumnsInRow() local
/external/webrtc/modules/third_party/fft/
Dfft.c158 int WebRtcIsac_Fftns(unsigned int ndim, const int dims[], in WebRtcIsac_Fftns()
/external/OpenCL-CTS/test_conformance/gl/
Dtest_images_read_common.cpp297 int ndim = 1; in test_cl_image_read() local
/external/tensorflow/tensorflow/core/kernels/
Dtile_ops.cc521 #define HANDLE_CASE(device, T, dtype, Tmultiples, ndim) \ argument
/external/tensorflow/tensorflow/python/ops/numpy_ops/
Dnp_array_ops.py966 def ndim(a): function
1164 def diag_indices(n, ndim=2): # pylint: disable=missing-docstring,redefined-outer-name argument
/external/python/cpython3/Modules/_ctypes/
Dctypes.h213 int ndim; member
/external/python/cpython2/Modules/_ctypes/
Dctypes.h253 int ndim; member
D_ctypes.c403 _ctypes_alloc_format_string_with_shape(int ndim, const Py_ssize_t *shape, in _ctypes_alloc_format_string_with_shape()
/external/python/cpython3/Include/cpython/
Dobject.h56 int ndim; member
/external/tensorflow/tensorflow/python/keras/
Dbackend.py1291 ndim=None, argument
1464 def ndim(x): function
/external/python/cpython2/Include/
Dobject.h171 int ndim; member

12