Home
last modified time | relevance | path

Searched defs:SetDims (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/c10/core/
H A DTensorImpl.h2483 bool SetDims(ArrayRef<int64_t> s) { in SetDims() function
2487 bool SetDims(ArrayRef<int> s) { in SetDims() function
2491 bool SetDims(ArrayRef<size_t> s) { in SetDims() function
2495 bool SetDims() { in SetDims() function
2499 bool SetDims(const int64_t d0) { in SetDims() function
2503 bool SetDims(const int64_t d0, const int64_t d1) { in SetDims() function
2507 bool SetDims(const int64_t d0, const int64_t d1, const int64_t d2) { in SetDims() function
2511 bool SetDims( in SetDims() function
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsplit_op.cc105 std::tuple<IndexType, IndexType, IndexType> SetDims( in SetDims() function in tensorflow::SplitOpBase
H A Dsplit_v_op.cc172 std::tuple<IndexType, IndexType, IndexType> SetDims( in SetDims() function in tensorflow::SplitVOpBase