Home
last modified time | relevance | path

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

1234567

/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A 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 …]
/aosp_15_r20/external/XNNPACK/src/operators/
H A 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()
H A 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()
/aosp_15_r20/external/tensorflow/tensorflow/java/src/main/native/
H A 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
/aosp_15_r20/external/XNNPACK/test/
H A Dtranspose-operator-tester.h24 const size_t num_dims, in reference_index()
38 inline TransposeOperatorTester& num_dims(size_t num_dims) { in num_dims() function
44 inline size_t num_dims() const { return this->num_dims_; } in num_dims() function
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A 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
H A Dpooling_ops.cc82 int num_dims() const { return num_spatial_dims_ + 2; } in num_dims() function in tensorflow::__anona9add1350111::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::__anona9add1350111::MaxPoolGradOp
468 int num_dims() const { return num_spatial_dims_ + 2; } in num_dims() function in tensorflow::__anona9add1350111::AvgPoolGradOp
556 int num_dims() const { return num_spatial_dims_ + 2; } in num_dims() function in tensorflow::__anona9add1350111::MaxPoolGradGradOp
/aosp_15_r20/external/tensorflow/tensorflow/c/
H A 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()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/xnnpack/
H A Dtranspose_tester.h36 inline TransposeTester& num_dims(int32_t num_dims) { in num_dims() argument
42 inline int32_t num_dims() const { return this->num_dims_; } in num_dims() function
H A Dtest_util.cc41 const int32_t num_dims = shape.size(); in QuantizeInt8PerChannel() local
78 const int32_t num_dims = shape.size(); in GetInt8QuantizationScalePerChannel() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/
H A 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
/aosp_15_r20/external/libopus/dnn/torch/dnntools/dnntools/relegance/
H A 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
/aosp_15_r20/external/XNNPACK/src/
H A 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()
H A Dnormalization.c41 size_t num_dims, in remove_dimension()
67 const size_t num_dims, in xnn_normalize_transpose_permutation()
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Dc_api_unified_experimental_test.cc124 int num_dims = sizeof(dims) / sizeof(dims[0]); in TEST_P() local
197 int num_dims = sizeof(dims) / sizeof(dims[0]); in TEST_P() local
283 int num_dims = sizeof(dims) / sizeof(dims[0]); in TEST_P() local
506 int num_dims = sizeof(dims) / sizeof(dims[0]); in TEST_P() local
854 int num_dims = sizeof(dims) / sizeof(dims[0]); in TEST_P() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A 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
/aosp_15_r20/external/executorch/backends/apple/mps/serialization/
H A Dmps_graph_schema.py444 num_dims: int = 0 variable in MPSMean
489 num_dims: int = 0 variable in MPSPermute
495 num_dims: int = 0 variable in MPSView
501 num_dims: int = 0 variable in MPSExpand
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A 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()
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/
H A Dcustom_device.cc23 int num_dims; in Shape() local
34 int num_dims; in NumElements() local
/aosp_15_r20/external/ComputeLibrary/src/common/utils/
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/flex/
H A Dutil.cc52 int num_dims = src.dims(); in CopyShapeAndType() local
225 int num_dims = tflite_tensor->dims->size; in CreateTfTensorFromTfLiteTensor() local
/aosp_15_r20/external/armnn/src/backends/neon/workloads/
H A 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
/aosp_15_r20/external/armnn/src/backends/cl/workloads/
H A 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

1234567