Home
last modified time | relevance | path

Searched refs:npy_intp (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/python/lib/core/
Dbfloat16.cc471 void NPyBfloat16_CopySwapN(void* dstv, npy_intp dstride, void* srcv, in NPyBfloat16_CopySwapN()
472 npy_intp sstride, npy_intp n, int swap, void* arr) { in NPyBfloat16_CopySwapN()
479 for (npy_intp i = 0; i < n; i++) { in NPyBfloat16_CopySwapN()
487 for (npy_intp i = 0; i < n; i++) { in NPyBfloat16_CopySwapN()
509 int NPyBfloat16_Fill(void* buffer_raw, npy_intp length, void* ignored) { in NPyBfloat16_Fill()
513 for (npy_intp i = 2; i < length; ++i) { in NPyBfloat16_Fill()
519 void NPyBfloat16_DotFunc(void* ip1, npy_intp is1, void* ip2, npy_intp is2, in NPyBfloat16_DotFunc()
520 void* op, npy_intp n, void* arr) { in NPyBfloat16_DotFunc()
524 for (npy_intp i = 0; i < n; ++i) { in NPyBfloat16_DotFunc()
547 int NPyBfloat16_ArgMaxFunc(void* data, npy_intp n, npy_intp* max_ind, in NPyBfloat16_ArgMaxFunc()
[all …]
Dndarray_tensor_bridge.h42 Status ArrayFromMemory(int dim_size, npy_intp* dims, void* data, DataType dtype,
Dndarray_tensor.cc307 gtl::InlinedVector<npy_intp, 4>* dims, in GetPyArrayDimensionsForTensor() argument
406 gtl::InlinedVector<npy_intp, 4> dims; in TF_TensorToMaybeAliasedPyArray()
425 gtl::InlinedVector<npy_intp, 4> dims; in TF_TensorToPyArray()
Dndarray_tensor_bridge.cc189 Status ArrayFromMemory(int dim_size, npy_intp* dims, void* data, DataType dtype, in ArrayFromMemory()
Dpy_func.cc134 std::array<npy_intp, 0> indices; in IsSingleNone()
/external/tensorflow/tensorflow/lite/python/interpreter_wrapper/
Dinterpreter_wrapper.cc85 PyObject* PyArrayFromFloatVector(const float* data, npy_intp size) { in PyArrayFromFloatVector()
93 PyObject* PyArrayFromIntVector(const int* data, npy_intp size) { in PyArrayFromIntVector()
620 std::vector<npy_intp> dims(tensor->dims->data, in GetTensor()
637 std::vector<npy_intp> sparse_buffer_dims(1); in GetTensor()
694 std::vector<npy_intp> dims(tensor->dims->data, in tensor()