Searched defs:NumDims (Results 1 – 9 of 9) sorted by relevance
121 static const int NumDims = internal::traits<OutputBackward>::NumDimensions; variable374 static const int NumDims = internal::traits<Input>::NumDimensions; variable
112 static const int NumDims = internal::traits<OutputBackward>::NumDimensions; variable412 static const int NumDims = internal::traits<Input>::NumDimensions; variable
111 const size_t NumDims = tensor.impl().dimensions().size(); in TensorContractionInputMapper() local138 const int NumDims = tensor.impl().impl().dimensions().size(); in TensorContractionInputMapper() local
111 const int NumDims = tensor.impl().dimensions().size(); in TensorContractionInputMapper() local157 const int NumDims = tensor.impl().impl().dimensions().size(); in TensorContractionInputMapper() local
33 static const int NumDims = NumInputDims + 1; member
115 const int NumDims = internal::traits<Input>::NumDimensions; variable
118 const int NumDims = internal::traits<Input>::NumDimensions; variable
187 Status TensorHandle::NumDims(int* num_dims) { in NumDims() function in tensorflow::TensorHandle
240 const std::size_t NumDims = sizeof...(otherIndices) + 2; in operator() local