/external/tensorflow/tensorflow/core/util/ |
D | tensor_format.h | 111 inline int GetTensorSpatialDims(int num_dims, TensorFormat format) { in GetTensorSpatialDims() 126 inline int GetFilterTensorSpatialDims(int num_dims, FilterTensorFormat format) { in GetFilterTensorSpatialDims() 162 inline int GetTensorBatchDimIndex(int num_dims, TensorFormat format) { in GetTensorBatchDimIndex() 182 inline int GetTensorFeatureDimIndex(int num_dims, TensorFormat format) { in GetTensorFeatureDimIndex() 200 inline int GetTensorInnerFeatureDimIndex(int num_dims, TensorFormat format) { in GetTensorInnerFeatureDimIndex() 206 inline int GetTensorInnerWidthDimIndex(int num_dims, TensorFormat format) { in GetTensorInnerWidthDimIndex() 215 inline int GetTensorSpatialDimIndex(int num_dims, TensorFormat format, in GetTensorSpatialDimIndex() 236 inline int GetFilterTensorSpatialDimIndex(int num_dims, in GetFilterTensorSpatialDimIndex() 254 int num_dims, FilterTensorFormat format) { in GetFilterTensorInnerInputChannelsDimIndex() 262 inline int GetFilterTensorInputChannelsDimIndex(int num_dims, in GetFilterTensorInputChannelsDimIndex() [all …]
|
D | padding.cc | 42 int num_dims, TensorFormat data_format) { in CheckValidPadding()
|
D | tensor_format_test.cc | 175 int num_dims = GetTensorDimsFromSpatialDims(num_spatial_dims, format); in RunDimensionIndexesTest() local 191 int num_dims = GetFilterTensorDimsFromSpatialDims(num_spatial_dims, format); in RunDimensionIndexesTest() local
|
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
D | conv_op_helpers.cc | 47 int num_dims = shape.dimensions_size(); in ExpandedFilterShapeForDepthwiseConvolution() local 168 const int num_dims = attrs.num_spatial_dims + 2; in CheckConvAttrs() local 257 int num_dims = attrs.num_spatial_dims + 2; in MakeXlaForwardConvOp() local 328 int num_dims = attrs.num_spatial_dims + 2; in MakeXlaBackpropInputConvOp() local 437 int num_dims = attrs.num_spatial_dims + 2; in MakeXlaBackpropFilterConvOp() local
|
D | pooling_ops.cc | 72 int num_dims() const { return num_spatial_dims_ + 2; } in num_dims() function in tensorflow::__anon3074d43f0111::PoolingOp 137 int num_dims = num_spatial_dims + 2; in XlaTensorFormat() local 277 int num_dims() const { return num_spatial_dims_ + 2; } in num_dims() function in tensorflow::__anon3074d43f0111::MaxPoolGradOp 401 int num_dims() const { return num_spatial_dims_ + 2; } in num_dims() function in tensorflow::__anon3074d43f0111::AvgPoolGradOp 484 int num_dims() const { return num_spatial_dims_ + 2; } in num_dims() function in tensorflow::__anon3074d43f0111::MaxPoolGradGradOp
|
D | reshape_op.cc | 43 const int64 num_dims = sizes_shape.num_elements(); in Compile() local
|
D | sparse_to_dense_op.cc | 39 const int64 num_dims = in Compile() local
|
/external/tensorflow/tensorflow/java/src/main/native/ |
D | tensor_jni.cc | 331 int num_dims = static_cast<int>(env->GetArrayLength(shape)); in Java_org_tensorflow_Tensor_allocate() local 391 size_t nonScalarTF_STRINGTensorSize(JNIEnv* env, jarray value, int num_dims) { in nonScalarTF_STRINGTensorSize() 414 void fillNonScalarTF_STRINGTensorData(JNIEnv* env, jarray value, int num_dims, in fillNonScalarTF_STRINGTensorData() 440 const int num_dims = static_cast<int>(env->GetArrayLength(shape)); in Java_org_tensorflow_Tensor_allocateNonScalarBytes() local 511 const jsize num_dims = TF_NumDims(t); in Java_org_tensorflow_Tensor_shape() local 527 int num_dims = TF_NumDims(t); in Java_org_tensorflow_Tensor_setValue() local 598 int num_dims = TF_NumDims(t); in Java_org_tensorflow_Tensor_readNDArray() local
|
/external/tensorflow/tensorflow/core/kernels/ |
D | sparse_sparse_binary_op_shared.cc | 66 typename TTypes<T>::ConstFlat b_values, int64 b_nnz, int num_dims, in UnionSparseIndicesAndValues() 183 const int num_dims = a_indices_t->dim_size(1); in Compute() local
|
D | roll_op.cc | 40 const int num_dims, const gtl::ArraySlice<int>& dim_size, in DoRoll() 103 const int num_dims, const gtl::ArraySlice<int>& dim_size, in DoRollWithMemcpy() 252 const int num_dims = input.dims(); in Compute() local
|
D | diag_op.cc | 49 const int num_dims = diagonal.dims(); in Compute() local 79 const int num_dims = tensor.dims(); in Compute() local
|
D | reshape_op.h | 98 const int64 num_dims = sizes.NumElements(); in ValidateSizes() local
|
D | clustering_ops_test.cc | 50 Graph* SetUpKmeansPlusPlusInitialization(int num_dims, int num_points, in SetUpKmeansPlusPlusInitialization() 175 Graph* SetUpNearestNeighbors(int num_dims, int num_points, int num_centers, in SetUpNearestNeighbors()
|
D | attention_ops.cc | 69 const int32 num_dims = input_shape.dims(); in Compute() local
|
D | sparse_slice_grad_op.cc | 72 const int num_dims = input_indices->dim_size(1); in Compute() local
|
D | sparse_to_dense_op.cc | 60 const int64 num_dims = indices.dims() > 1 ? indices.dim_size(1) : 1; in Compute() local
|
D | sparse_add_grad_op.cc | 67 const int num_dims = a_indices->dim_size(1); in Compute() local
|
D | nth_element_op.cc | 52 const int num_dims = input_in.dims(); in Compute() local
|
/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
D | svd.cc | 118 const int64 num_dims = a_shape.rank(); in HouseRow() local 184 const int64 num_dims = a_shape.rank(); in HouseCol() local 262 const int64 num_dims = a_shape.rank(); in HouseHolderBidiagonalization() local 467 const int64 num_dims = d_shape.rank(); in OneSidedJacobiUpdate() local 577 const int64 num_dims = shape.rank(); in ComputeFrobeniusNorms() local 732 const int64 num_dims = shape.rank(); in SortBySingularValuesAndPostProcessing() local 836 const int64 num_dims = a_shape.rank(); in SVD() local
|
D | self_adjoint_eig.cc | 122 const int64 num_dims = w_shape.rank(); in Update() local 198 const int64 num_dims = shape.rank(); in ComputeFrobeniusNorms() local 343 const int64 num_dims = shape.rank(); in SortByEigenvalues() local 407 const int64 num_dims = a_shape.rank(); in SelfAdjointEig() local
|
/external/tensorflow/tensorflow/python/kernel_tests/ |
D | reduce_join_op_test.py | 34 def _input_array(num_dims): argument 48 def _joined_array(num_dims, reduce_dim): argument
|
/external/tensorflow/tensorflow/lite/delegates/flex/ |
D | util.cc | 40 int num_dims = src.dims(); in CopyShapeAndType() local
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | parallel_loop_emitter.cc | 41 const int64 num_dims = shape_.dimensions_size(); in EmitIndexAndSetExitBasicBlock() local
|
D | conv_canonicalization.cc | 46 const int64 num_dims = num_spatial_dims + 2; in Run() local
|
/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/ |
D | scatter_add_ndim_op.cc | 69 const int32 num_dims = in Compute() local
|