| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_format.h | 116 inline int GetTensorSpatialDims(int num_dims, TensorFormat format) { in GetTensorSpatialDims()
134 inline int GetFilterTensorSpatialDims(int num_dims, FilterTensorFormat format) { in GetFilterTensorSpatialDims()
173 inline int GetTensorBatchDimIndex(int num_dims, TensorFormat format) { in GetTensorBatchDimIndex()
193 inline int GetTensorFeatureDimIndex(int num_dims, TensorFormat format) { in GetTensorFeatureDimIndex()
211 inline int GetTensorInnerFeatureDimIndex(int num_dims, TensorFormat format) { in GetTensorInnerFeatureDimIndex()
217 inline int GetTensorInnerWidthDimIndex(int num_dims, TensorFormat format) { in GetTensorInnerWidthDimIndex()
226 inline int GetTensorSpatialDimIndex(int num_dims, TensorFormat format, in GetTensorSpatialDimIndex()
247 inline int GetFilterTensorSpatialDimIndex(int num_dims, in GetFilterTensorSpatialDimIndex()
265 int num_dims, FilterTensorFormat format) { in GetFilterTensorInnerInputChannelsDimIndex()
273 inline int GetFilterTensorInputChannelsDimIndex(int num_dims, in GetFilterTensorInputChannelsDimIndex()
[all …]
|
| D | padding.cc | 38 int num_dims, TensorFormat data_format) { in CheckValidPadding()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | gradients_util.cc | 55 int64_t dims[], int num_dims, in TensorHandleWithDimsFloatHelper()
69 int64_t dims[], int num_dims, in TensorHandleWithDimsIntHelper()
99 int64_t dims[], int num_dims, in TensorHandleWithDimsFloat()
116 int num_dims, AbstractTensorHandle** tensor) { in TensorHandleWithDimsInt()
142 int num_dims) { in GetTensorHandleUtilFloat()
153 int64_t dims[], int num_dims) { in GetTensorHandleUtilInt()
|
| D | mnist_gradients_test.cc | 82 int num_dims = 2; in TEST_P() local
152 int num_dims = 2; in TEST_P() local
228 int num_dims = 2; in TEST_P() local
308 int num_dims = 2; in TEST_P() local
366 int num_dims = 2; in TEST_P() local
468 int num_dims = 2; in TEST_P() local
520 int num_dims = 2; in TEST_P() local
|
| D | c_api_unified_experimental.h | 69 int num_dims; // Must be >= -1; -1 represents unknown rank. member
|
| D | gradient_checker.cc | 40 int num_dims = TF_NumDims(t); in GetDims() local
117 int num_dims = TF_NumDims(theta_tensor); in CalcNumericalGrad() local
|
| D | unified_api_testutil.cc | 147 int64_t* dims, int num_dims, in TestTensorHandleWithDimsFloat()
162 int64_t* dims, int num_dims, in TestTensorHandleWithDimsInt()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | custom_device.cc | 21 int num_dims; in Shape() local
32 int num_dims; in NumElements() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | conv_op_helpers.cc | 69 int num_dims = filter_shape.dimensions_size(); in TransposeFilterForGroupConvolutionBackpropInput() local
100 const int num_dims = attrs.num_spatial_dims + 2; in CheckConvAttrs() local
198 int num_dims = attrs.num_spatial_dims + 2; in MakeXlaForwardConvOp() local
299 int num_dims = attrs.num_spatial_dims + 2; in MakeXlaBackpropInputConvOp() local
429 int num_dims = attrs.num_spatial_dims + 2; in MakeXlaBackpropFilterConvOp() local
|
| D | pooling_ops.cc | 74 int num_dims() const { return num_spatial_dims_ + 2; } in num_dims() function in tensorflow::__anon9297e9860111::PoolingOp
139 int num_dims = num_spatial_dims + 2; in XlaTensorFormat() local
284 int num_dims() const { return num_spatial_dims_ + 2; } in num_dims() function in tensorflow::__anon9297e9860111::MaxPoolGradOp
422 int num_dims() const { return num_spatial_dims_ + 2; } in num_dims() function in tensorflow::__anon9297e9860111::AvgPoolGradOp
504 int num_dims() const { return num_spatial_dims_ + 2; } in num_dims() function in tensorflow::__anon9297e9860111::MaxPoolGradGradOp
|
| /external/tensorflow/tensorflow/java/src/main/native/ |
| D | tensor_jni.cc | 291 int num_dims = static_cast<int>(env->GetArrayLength(shape)); in Java_org_tensorflow_Tensor_allocate() local
350 void checkForNullEntries(JNIEnv* env, jarray value, int num_dims) { in checkForNullEntries()
364 void fillNonScalarTF_STRINGTensorData(JNIEnv* env, jarray value, int num_dims, in fillNonScalarTF_STRINGTensorData()
390 const int num_dims = static_cast<int>(env->GetArrayLength(shape)); in Java_org_tensorflow_Tensor_allocateNonScalarBytes() local
460 const jsize num_dims = TF_NumDims(t); in Java_org_tensorflow_Tensor_shape() local
476 int num_dims = TF_NumDims(t); in Java_org_tensorflow_Tensor_setValue() local
535 int num_dims = TF_NumDims(t); in Java_org_tensorflow_Tensor_readNDArray() local
|
| /external/tensorflow/tensorflow/lite/tools/versioning/ |
| D | op_version.h | 60 int32_t num_dims; member
63 int32_t num_dims; member
68 int32_t num_dims; member
|
| /external/XNNPACK/src/operators/ |
| D | constant-pad-nd.c | 72 size_t num_dims, in setup_constant_pad_nd()
188 size_t num_dims, in xnn_setup_constant_pad_nd_x32()
|
| /external/XNNPACK/src/subgraph/ |
| D | static-reshape.c | 19 size_t num_dims, in xnn_define_static_reshape()
|
| D | static-constant-pad.c | 53 const size_t num_dims = subgraph->values[input_id].shape.num_dims; in xnn_define_static_constant_pad() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | svd.cc | 113 const int64 num_dims = a_shape.rank(); in HouseRow() local
178 const int64 num_dims = a_shape.rank(); in HouseCol() local
256 const int64 num_dims = a_shape.rank(); in HouseHolderBidiagonalization() local
458 const int64 num_dims = d_shape.rank(); in OneSidedJacobiUpdate() local
568 auto num_dims = static_cast<int32>(shape.rank()); in ComputeToleranceComparison() local
722 const int64 num_dims = shape.rank(); in SortBySingularValuesAndPostProcessing() local
823 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
|
| D | qr.cc | 60 const int num_dims = a_shape.rank(); in QRDecomposition() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | roll_op.cc | 63 const int num_dims = input.dims(); in Compute() local
122 const int num_dims, const gtl::ArraySlice<int32>& dim_size, in DoRoll()
185 const int num_dims, in DoRollWithMemcpy()
|
| 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 | diag_op.cc | 49 const int num_dims = diagonal.dims(); in Compute() local
79 const int num_dims = tensor.dims(); in Compute() 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/XNNPACK/src/ |
| D | tensor.c | 21 size_t num_dims, in xnn_define_tensor_value()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | util.cc | 40 int num_dims = src.dims(); in CopyShapeAndType() local
|
| /external/tensorflow/tensorflow/c/ |
| D | tf_tensor.cc | 71 const int64_t* dims, int num_dims, size_t len) { in CreateTensor()
91 int num_dims, size_t len) { in TF_AllocateTensor()
100 TF_Tensor* TF_NewTensor(TF_DataType dtype, const int64_t* dims, int num_dims, in TF_NewTensor()
|