Home
last modified time | relevance | path

Searched defs:num_dims (Results 1 – 25 of 164) sorted by relevance

1234567

/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dtranspose_tester.h36 inline TransposeTester& num_dims(int32_t num_dims) { in num_dims() function
42 inline int32_t num_dims() const { return this->num_dims_; } in num_dims() function
Dtest_util.cc41 const int32_t num_dims = shape.size(); in QuantizeInt8PerChannel() local
78 const int32_t num_dims = shape.size(); in GetInt8QuantizationScalePerChannel() local
/external/tensorflow/tensorflow/core/util/
Dtensor_format.h116 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 …]
/external/XNNPACK/test/
Dtranspose-operator-tester.h24 const size_t num_dims, in reference_index()
38 inline TransposeOperatorTester& num_dims(size_t num_dims) { in num_dims() argument
44 inline size_t num_dims() const { return this->num_dims_; } in num_dims() function
Dtranspose-normalization-tester.h16 inline TransposeNormalizationTester& num_dims(size_t num_dims) { in num_dims() argument
22 inline size_t num_dims() const { return this->num_dims_; } in num_dims() function
/external/XNNPACK/src/operators/
Dconstant-pad-nd.c93 size_t num_dims, in setup_constant_pad_nd()
210 size_t num_dims, in xnn_setup_constant_pad_nd_x8()
227 size_t num_dims, in xnn_setup_constant_pad_nd_x16()
244 size_t num_dims, in xnn_setup_constant_pad_nd_x32()
Dtranspose-nd.c24 size_t num_dims, in reorder_array()
107 const size_t num_dims, in setup_transpose_nd()
355 size_t num_dims, in xnn_setup_transpose_nd_x32()
378 size_t num_dims, in xnn_setup_transpose_nd_x16()
401 size_t num_dims, in xnn_setup_transpose_nd_x8()
424 const size_t num_dims, in run_transpose_nd()
471 const size_t num_dims, in xnn_run_transpose_nd_x32()
493 const size_t num_dims, in xnn_run_transpose_nd_x16()
515 const size_t num_dims, in xnn_run_transpose_nd_x8()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcustom_device.cc23 int num_dims; in Shape() local
34 int num_dims; in NumElements() local
/external/libopus/dnn/torch/dnntools/dnntools/relegance/
Drelegance.py169 def estimate_conv_parameters(model, num_channels, num_dims, width, max_stride=10): argument
227 def inspect_conv_model(model, num_channels, num_dims, max_width=10000, width_hint=None, stride_hint… argument
334 def relegance_create_tconv_kernel(model, num_channels, num_dims, width_hint=None, stride_hint=None,… argument
/external/XNNPACK/src/
Dtensor.c22 size_t num_dims, in xnn_define_tensor_value()
81 size_t num_dims, in xnn_define_quantized_tensor_value()
170 size_t num_dims, in xnn_define_channelwise_quantized_tensor_value()
Dnormalization.c41 size_t num_dims, in remove_dimension()
67 const size_t num_dims, in xnn_normalize_transpose_permutation()
/external/ComputeLibrary/src/common/utils/
DLegacySupport.cpp73 const auto num_dims = info.num_dimensions(); in create_tensor_shape_array() local
99 const auto num_dims = info.num_dimensions(); in convert_to_descriptor() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dconv_op_helpers.cc69 int num_dims = filter_shape.dimensions_size(); in TransposeFilterForGroupConvolutionBackpropInput() local
100 const int num_dims = attrs.num_spatial_dims + 2; in CheckConvAttrs() local
197 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
Dpooling_ops.cc82 int num_dims() const { return num_spatial_dims_ + 2; } in num_dims() function in tensorflow::__anon26e17bb50111::PoolingOp
147 int num_dims = num_spatial_dims + 2; in XlaTensorFormat() local
332 int num_dims() const { return num_spatial_dims_ + 2; } in num_dims() function in tensorflow::__anon26e17bb50111::MaxPoolGradOp
468 int num_dims() const { return num_spatial_dims_ + 2; } in num_dims() function in tensorflow::__anon26e17bb50111::AvgPoolGradOp
556 int num_dims() const { return num_spatial_dims_ + 2; } in num_dims() function in tensorflow::__anon26e17bb50111::MaxPoolGradGradOp
/external/tensorflow/tensorflow/core/kernels/
Dsparse_slice_grad_op.cc39 const int num_dims = input_indices_mat.dimension(1); in operator ()() local
122 const int num_dims = input_indices->dim_size(1); in Compute() local
Droll_op.cc63 const int num_dims = input.dims(); in Compute() local
122 const int num_dims, const gtl::ArraySlice<int32> dim_size, in DoRoll()
185 const int64_t num_elements, const int num_dims, in DoRollWithMemcpy()
Dsparse_sparse_binary_op_shared.cc66 typename TTypes<T>::ConstFlat b_values, int64_t b_nnz, int num_dims, in UnionSparseIndicesAndValues()
170 const int num_dims = a_indices_t->dim_size(1); in Compute() local
/external/tensorflow/tensorflow/c/
Dtf_tensor.cc71 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()
151 void TF_SetShape(TF_Tensor* t, const int64_t* dims, int num_dims) { in TF_SetShape()
238 void TensorInterface::SetShape(const int64_t* dims, int num_dims) { in SetShape()
/external/tensorflow/tensorflow/java/src/main/native/
Dtensor_jni.cc293 int num_dims = static_cast<int>(env->GetArrayLength(shape)); in Java_org_tensorflow_Tensor_allocate() local
352 void checkForNullEntries(JNIEnv* env, jarray value, int num_dims) { in checkForNullEntries()
367 void fillNonScalarTF_STRINGTensorData(JNIEnv* env, jarray value, int num_dims, in fillNonScalarTF_STRINGTensorData()
394 const int num_dims = static_cast<int>(env->GetArrayLength(shape)); in Java_org_tensorflow_Tensor_allocateNonScalarBytes() local
464 const jsize num_dims = TF_NumDims(t); in Java_org_tensorflow_Tensor_shape() local
480 int num_dims = TF_NumDims(t); in Java_org_tensorflow_Tensor_setValue() local
539 int num_dims = TF_NumDims(t); in Java_org_tensorflow_Tensor_readNDArray() local
/external/tensorflow/tensorflow/lite/delegates/flex/
Dutil.cc52 int num_dims = src.dims(); in CopyShapeAndType() local
225 int num_dims = tflite_tensor->dims->size; in CreateTfTensorFromTfLiteTensor() local
/external/armnn/src/backends/neon/workloads/
DNeonWorkloadUtils.hpp139 unsigned int num_dims = static_cast<unsigned int>(m_begin.size()); in SetNeonStridedSliceData() local
161 unsigned int num_dims = static_cast<unsigned int>(m_begin.size()); in SetNeonSliceData() local
/external/armnn/src/backends/cl/workloads/
DClWorkloadUtils.hpp79 unsigned int num_dims = static_cast<unsigned int>(m_begin.size()); in SetClStridedSliceData() local
100 unsigned int num_dims = static_cast<unsigned int>(m_begin.size()); in SetClSliceData() local
/external/pytorch/torch/csrc/jit/tensorexpr/
Dblock_codegen.cpp155 auto num_dims = block_analysis_->getMultiDimBuf(buf)->dims().size(); in PrintTensorInfo() local
169 auto num_dims = block_analysis_->getMultiDimBuf(buf)->dims().size(); in PrintTensorInfo() local
185 auto num_dims = multidimbuf->dims().size(); in PrintArguments() local
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dsvd.cc114 const int64_t num_dims = a_shape.rank(); in HouseRow() local
179 const int64_t num_dims = a_shape.rank(); in HouseCol() local
257 const int64_t num_dims = a_shape.rank(); in HouseHolderBidiagonalization() local
459 const int64_t num_dims = d_shape.rank(); in OneSidedJacobiUpdate() local
569 auto num_dims = static_cast<int32_t>(shape.rank()); in ComputeToleranceComparison() local
737 const int64_t num_dims = shape.rank(); in SortBySingularValuesAndPostProcessing() local
838 const int64_t num_dims = a_shape.rank(); in SVD() local
/external/tensorflow/tensorflow/c/eager/
Dc_api_unified_experimental.h69 int num_dims; // Must be >= -1; -1 represents unknown rank. member

1234567