| /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
|