| /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 | tile_functor_gpu.h | 35 const int32 ndims, T* dst) { 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 | transpose_functor_gpu.cu.cc | 36 const int32 ndims, T* dst) { in TransposeKernel()
63 const int32 ndims = in.dims(); in TransposeSimple() local
|
| D | tile_functor_cpu.cc | 29 const int ndims = in.dims(); in TileSimple() local
|
| D | cwise_ops_common.h | 71 int ndims; member
102 const int ndims = state.ndims; in Compute() local
|
| D | sparse_dense_binary_op_shared.cc | 120 const int ndims = lhs_dims.size(); in Compute() local
|
| D | self_adjoint_eig_v2_op_gpu.cc | 50 const int ndims = input.dims(); in ComputeAsync() local
|
| D | betainc_op.cc | 88 int ndims = static_cast<int>(a_shaper.x_reshape().size()); in Compute() local
|
| D | cholesky_op.cc | 102 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 | 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 | 240 const int ndims = in.dims(); in DoMatrixTransposeImpl() local
|
| D | transpose_functor_cpu.cc | 37 const int ndims = in.dims(); in TransposeSimple() local
|
| D | matrix_inverse_op.cc | 100 const int ndims = input.dims(); in ComputeAsync() local
|
| D | ragged_tensor_to_sparse_kernel.cc | 130 int64 ndims = rt_nested_splits_len + rt_dense_values_in.dims(); in Compute() local
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_tensor_coding.cc | 83 const int ndims = val.shape().dims(); in SkeletonEncodingSizeUpperBound() local
96 const int ndims = val.shape().dims(); in EncodeSkeleton() local
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | scatter_nd_op_test.py | 35 def _FlatInnerDims(tensor, ndims=2): argument
42 def _FlatOuterDims(tensor, ndims=2): argument
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.cc | 238 BatchDescriptor::BatchDescriptor(int ndims) in BatchDescriptor()
383 FilterDescriptor::FilterDescriptor(int ndims) { in FilterDescriptor()
455 ConvolutionDescriptor::ConvolutionDescriptor(int ndims) { in ConvolutionDescriptor()
502 PoolingDescriptor::PoolingDescriptor(int ndims) in PoolingDescriptor()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | triangular_solve_expander.cc | 43 int ndims = shape.rank(); in DiagonalBlocks() local
263 int64 ndims = a_shape.rank(); in SolveWithInvertedDiagonalBlocks() local
364 const int64 ndims = a_shape.rank(); in BuildTriangularSolve() local
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_util.py | 43 def get_positive_axis(axis, ndims): argument
|
| /external/tensorflow/tensorflow/contrib/distributions/python/ops/ |
| D | sample_stats.py | 468 def _make_static_axis_non_negative(axis, ndims): argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_ops_test.py | 614 def _compare(self, sp_t, reduction_axes, ndims, keep_dims, do_sum): argument
660 def _compare_all(self, sp_t, reduction_axes, ndims): argument
732 def _testSparseReduceShape(self, sp_t, reduction_axes, ndims, keep_dims, argument
|
| D | scatter_nd_ops_test.py | 48 def _FlatInnerDims(tensor, ndims=2): argument
55 def _FlatOuterDims(tensor, ndims=2): argument
|
| /external/tensorflow/tensorflow/contrib/distributions/python/ops/bijectors/ |
| D | chain.py | 43 def _use_static_shape(input_tensor, ndims): argument
|