| /external/tensorflow/tensorflow/core/framework/ |
| D | ops_util.h | 90 const int ndims = shape.dims(); in ComputeStride() local
103 const int ndims = shape.rank(); in ComputeEigenStrides() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| 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 | broadcast_to_op.h | 75 const int ndims = bcast.y_reshape().size(); in operator() local
|
| D | tile_functor_cpu.h | 28 const int ndims = in.dims(); in TileSimpleImpl() 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 | cwise_ops_common.h | 67 int ndims; member
166 const int ndims = state.ndims; in Compute() local
|
| D | transpose_functor_cpu.cc | 37 const int ndims = in.dims(); in TransposeSimple() local
|
| D | sparse_dense_binary_op_shared.cc | 120 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
|
| D | sparse_tensor_dense_add_op.cc | 86 const int ndims = static_cast<int>(a_indices_t->dim_size(1)); in Compute() local
|
| D | transpose_functor.h | 242 const int ndims = in.dims(); in DoMatrixTransposeImpl() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/v1_compat_tests/ |
| D | scatter_nd_ops_test.py | 35 def _FlatInnerDims(tensor, ndims=2): argument
42 def _FlatOuterDims(tensor, ndims=2): argument
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_tensor_coding.cc | 85 const int ndims = val.shape().dims(); in SkeletonEncodingSizeUpperBound() local
98 const int ndims = val.shape().dims(); in EncodeSkeleton() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matrix_inverse_op.cc | 32 int64 ndims = input_shape.dims(); in Compile() 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/xla/service/ |
| D | cholesky_expander.cc | 59 const int ndims = a_shape.rank(); in CholeskyUnblocked() local
133 const int ndims = a_shape.rank(); in BuildCholesky() local
|
| D | triangular_solve_expander.cc | 44 int ndims = shape.rank(); in DiagonalBlocks() local
145 int64 ndims = a_shape.rank(); in SolveWithInvertedDiagonalBlocks() local
379 const int64 ndims = a_shape.rank(); in SolveByInvertingDiagonalBlocks() local
487 const int64 ndims = a_shape.rank(); in BuildTriangularSolve() local
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | scatter_nd_op_test.py | 36 def _FlatInnerDims(tensor, ndims=2): argument
43 def _FlatOuterDims(tensor, ndims=2): argument
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.cc | 269 BatchDescriptor::BatchDescriptor(int ndims) in BatchDescriptor()
414 FilterDescriptor::FilterDescriptor(int ndims) { in FilterDescriptor()
486 ConvolutionDescriptor::ConvolutionDescriptor(int ndims) { in ConvolutionDescriptor()
532 PoolingDescriptor::PoolingDescriptor(int ndims) in PoolingDescriptor()
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | tridiagonal_solve_op_gpu.cu.cc | 256 const int ndims = lhs.dims(); in Compute() local
285 const int ndims = lhs.dims(); in ComputeWithGtsvBatched() local
330 const int ndims = lhs.dims(); in TransposeLhsForGtsvBatched() local
|
| D | tridiagonal_matmul_op_gpu.cu.cc | 68 const int ndims = rhs.dims(); in Compute() local
|
| D | self_adjoint_eig_v2_op_gpu.cc | 50 const int ndims = input.dims(); in ComputeAsync() local
|
| D | determinant_op.cc | 135 const int ndims = input.dims(); in ComputeAsync() local
275 const int ndims = input.dims(); in ComputeAsync() local
|
| D | cholesky_op.cc | 101 const int ndims = input.dims(); in ComputeAsync() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_ops_test.py | 631 def _compare(self, sp_t, reduction_axes, ndims, keep_dims, do_sum): argument
677 def _compare_all(self, sp_t, reduction_axes, ndims): argument
747 def _testSparseReduceShape(self, sp_t, reduction_axes, ndims, keep_dims, argument
|