| /external/tensorflow/tensorflow/core/framework/ |
| D | ops_util.h | 91 const int ndims = shape.dims(); in ComputeStride() local
104 const int ndims = shape.rank(); in ComputeEigenStrides() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_utils.cc | 168 int64_t ndims = indices.dim_size(1); in ValidateSparseTensorShape() local
185 const int64_t ndims = indices.dimension(1); in CreateIndexString() local
205 int64_t ndims = indices.dim_size(1); in ValidateSparseTensorIndicesUnordered() local
229 int64_t ndims = indices.dim_size(1); in ValidateSparseTensorIndicesOrdered() local
|
| D | tile_functor_gpu.h | 34 const int32* __restrict__ buf, const int32 ndims, in TileKernel()
58 const int32 ndims = in.dims(); in TileSimple() local
|
| D | sparse_reorder_op_gpu.cu.cc | 34 const int64 ndims, in IndicesFlattenKernel()
51 const int64 nnz, const int64 ndims, const int64* __restrict__ permutation, in PermuteIndicesAndValuesKernel()
|
| D | sparse_tensor_dense_add_op.cc | 52 int64_t ndims = a_indices->dim_size(1); in ValidateInputs() local
116 const int ndims = static_cast<int>(a_indices_t->dim_size(1)); in Compute() local
|
| D | broadcast_to_op.h | 62 const int ndims = bcast.y_reshape().size(); in operator() local
|
| D | transpose_functor_gpu.cu.cc | 37 const int32 ndims, T* __restrict__ dst) { in TransposeKernel()
64 const int32 ndims = in.dims(); in TransposeSimple() local
|
| D | tile_functor_cpu.h | 28 const int ndims = in.dims(); in TileSimpleImpl() local
|
| D | cwise_ops_common.h | 67 int ndims; member
176 const int ndims = state.ndims; in Compute() local
|
| D | sparse_to_dense_op_gpu.cu.cc | 40 const int ndims, T* __restrict__ dense) { in SparseToDenseKernel()
65 const int ndims, int* __restrict__ status) { in CheckIndicesValid()
|
| D | transpose_functor_cpu.cc | 37 const int ndims = in.dims(); in TransposeSimple() local
|
| D | sparse_dense_binary_op_shared.cc | 143 const int ndims = lhs_dims.size(); in Compute() local
|
| D | betainc_op.cc | 88 int ndims = static_cast<int>(a_shaper.x_reshape().size()); in Compute() local
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | tridiagonal_matmul_op_gpu.cu.cc | 68 const int ndims = rhs.dims(); in Compute() local
99 const int ndims = rhs.dims(); in ValidateInputTensor() local
|
| D | tridiagonal_solve_op_gpu.cu.cc | 257 const int ndims = lhs.dims(); in Compute() local
286 const int ndims = lhs.dims(); in ComputeWithGtsvBatched() local
331 const int ndims = lhs.dims(); in TransposeLhsForGtsvBatched() local
|
| D | determinant_op.cc | 135 const int ndims = input.dims(); in ComputeAsync() local
275 const int ndims = input.dims(); in ComputeAsync() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/v1_compat_tests/ |
| D | scatter_nd_ops_test.py | 31 def _FlatInnerDims(tensor, ndims=2): argument
38 def _FlatOuterDims(tensor, ndims=2): argument
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_tensor_coding.cc | 82 const int ndims = val.shape().dims(); in SkeletonEncodingSizeUpperBound() local
95 const int ndims = val.shape().dims(); in EncodeSkeleton() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | lu_decomposition.cc | 30 const int ndims = a_shape.rank(); in LuDecomposition() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matrix_inverse_op.cc | 32 int64_t ndims = input_shape.dims(); in Compile() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | cholesky_expander.cc | 59 const int ndims = a_shape.rank(); in CholeskyUnblocked() local
131 const int ndims = a_shape.rank(); in BuildCholesky() local
|
| D | triangular_solve_expander.cc | 44 int ndims = shape.rank(); in DiagonalBlocks() local
139 int64_t ndims = a_shape.rank(); in SolveWithInvertedDiagonalBlocks() local
372 const int64_t ndims = a_shape.rank(); in SolveByInvertingDiagonalBlocks() local
480 const int64_t ndims = a_shape.rank(); in BuildTriangularSolve() local
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | scatter_nd_op_test.py | 32 def _FlatInnerDims(tensor, ndims=2): argument
39 def _FlatOuterDims(tensor, ndims=2): argument
|
| /external/ComputeLibrary/src/common/utils/ |
| D | LegacySupport.cpp | 62 TensorShape create_legacy_tensor_shape(int32_t ndims, int32_t *shape) in create_legacy_tensor_shape()
|
| /external/tensorflow/tensorflow/python/keras/layers/normalization/ |
| D | layer_normalization.py | 189 def _fused_can_be_used(self, ndims): argument
|