Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dmatrix.cc62 auto n_dims = static_cast<int32_t>(shape.rank()); in GetDiagonalMask() local
63 TF_RET_CHECK(n_dims >= 2); in GetDiagonalMask()
64 auto m = shape.dimensions(n_dims - 2); in GetDiagonalMask()
65 auto n = shape.dimensions(n_dims - 1); in GetDiagonalMask()
67 shape.dimensions().subspan(/*pos=*/0, /*len=*/n_dims - 2); in GetDiagonalMask()
80 auto n_dims = static_cast<int32_t>(shape.rank()); in GetMatrixDiagonal() local
81 TF_RET_CHECK(n_dims >= 2); in GetMatrixDiagonal()
82 const int64_t m = shape.dimensions(n_dims - 2); in GetMatrixDiagonal()
83 const int64_t n = shape.dimensions(n_dims - 1); in GetMatrixDiagonal()
87 zero_size_shape.DeleteDimension(n_dims - 1); in GetMatrixDiagonal()
[all …]
Dslicing.cc51 const int64_t n_dims = shape.rank(); in SliceInMinorDims() local
52 TF_RET_CHECK(n_minor_dims <= n_dims); in SliceInMinorDims()
55 /*len=*/n_dims - n_minor_dims); in SliceInMinorDims()
58 std::vector<int64_t> padded_start(n_dims, 0); in SliceInMinorDims()
63 std::vector<int64_t> padded_end(n_dims); in SliceInMinorDims()
67 std::vector<int64_t> strides(n_dims, 1); in SliceInMinorDims()
76 const int64_t n_dims = shape.rank(); in UpdateSlice() local
78 TF_RET_CHECK(start_size == n_dims); in UpdateSlice()
96 const int64_t n_dims = shape.rank(); in UpdateSliceInMinorDims() local
98 TF_RET_CHECK(n_minor_dims <= n_dims); in UpdateSliceInMinorDims()
[all …]
/external/eigen/doc/snippets/
DTutorial_Map_using.cpp4 const int n_dims = 5; variable
6 MatrixType m1(n_dims), m2(n_dims);
/external/ComputeLibrary/tests/validation/reference/
DSoftmaxLayer.cpp44 const int32_t n_dims = static_cast<int32_t>(src.shape().num_dimensions()); in softmax_layer_generic() local
45 ARM_COMPUTE_ERROR_ON(axis < -n_dims || axis >= n_dims); in softmax_layer_generic()
47 const unsigned int actual_axis = static_cast<unsigned int>(wrap_around(axis, n_dims)); in softmax_layer_generic()
/external/ComputeLibrary/include/libnpy/
Dnpy.hpp492 SaveArrayAsNumpy(const std::string &filename, bool fortran_order, unsigned int n_dims, const unsign… in SaveArrayAsNumpy() argument
502 std::vector <ndarray_len_t> shape_v(shape, shape + n_dims); in SaveArrayAsNumpy()
513 SaveArrayAsNumpy(const std::string &filename, bool fortran_order, unsigned int n_dims, const unsign… in SaveArrayAsNumpy() argument
515 SaveArrayAsNumpy(filename, fortran_order, n_dims, shape, data.data()); in SaveArrayAsNumpy()
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_dnn.cc1518 int n_dims; in CheckAndFetchProjectionWeights() local
1523 /*nbDims=*/&n_dims, /*filterDimA=*/dims)); in CheckAndFetchProjectionWeights()
1609 int n_dims; in Create() local
1614 /*nbDims=*/&n_dims, /*filterDimA=*/dims)); in Create()