Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/core/util/
Dtensor_format.h91 inline int GetTensorSpatialDims(int num_dims, TensorFormat format) { in GetTensorSpatialDims()
99 inline int GetFilterTensorSpatialDims(int num_dims, FilterTensorFormat format) { in GetFilterTensorSpatialDims()
130 inline int GetTensorBatchDimIndex(int num_dims, TensorFormat format) { in GetTensorBatchDimIndex()
145 inline int GetTensorFeatureDimIndex(int num_dims, TensorFormat format) { in GetTensorFeatureDimIndex()
159 inline int GetTensorInnerFeatureDimIndex(int num_dims, TensorFormat format) { in GetTensorInnerFeatureDimIndex()
165 inline int GetTensorSpatialDimIndex(int num_dims, TensorFormat format, in GetTensorSpatialDimIndex()
182 inline int GetFilterTensorSpatialDimIndex(int num_dims, in GetFilterTensorSpatialDimIndex()
200 int num_dims, FilterTensorFormat format) { in GetFilterTensorInnerInputChannelsDimIndex()
208 inline int GetFilterTensorInputChannelsDimIndex(int num_dims, in GetFilterTensorInputChannelsDimIndex()
223 inline int GetFilterTensorOutputChannelsDimIndex(int num_dims, in GetFilterTensorOutputChannelsDimIndex()
Dtensor_format_test.cc160 int num_dims = GetTensorDimsFromSpatialDims(num_spatial_dims, format); in RunDimensionIndexesTest() local
176 int num_dims = GetFilterTensorDimsFromSpatialDims(num_spatial_dims, format); in RunDimensionIndexesTest() local
/external/tensorflow/tensorflow/java/src/main/native/
Dtensor_jni.cc331 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()
408 void fillNonScalarTF_STRINGTensorData(JNIEnv* env, jarray value, int num_dims, in fillNonScalarTF_STRINGTensorData()
434 const int num_dims = static_cast<int>(env->GetArrayLength(shape)); in Java_org_tensorflow_Tensor_allocateNonScalarBytes() local
504 const jsize num_dims = TF_NumDims(t); in Java_org_tensorflow_Tensor_shape() local
520 int num_dims = TF_NumDims(t); in Java_org_tensorflow_Tensor_setValue() local
591 int num_dims = TF_NumDims(t); in Java_org_tensorflow_Tensor_readNDArray() local
/external/tensorflow/tensorflow/core/kernels/
Dsparse_sparse_binary_op_shared.cc66 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
Droll_op.cc39 const int num_dims, const gtl::ArraySlice<int>& dim_size, in DoRoll()
102 const int num_dims, const gtl::ArraySlice<int>& dim_size, in DoRollWithMemcpy()
251 const int num_dims = input.dims(); in Compute() local
Ddiag_op.cc49 const int num_dims = diagonal.dims(); in Compute() local
79 const int num_dims = tensor.dims(); in Compute() local
Dreshape_op.h98 const int64 num_dims = sizes.NumElements(); in ValidateSizes() local
Dattention_ops.cc45 const int32 num_dims = input_shape.dims(); in Compute() local
Dsparse_to_dense_op.cc60 const int64 num_dims = indices.dims() > 1 ? indices.dim_size(1) : 1; in Compute() local
Dsparse_add_grad_op.cc67 const int num_dims = a_indices->dim_size(1); in Compute() local
Dnth_element_op.cc52 const int num_dims = input_in.dims(); in Compute() local
Dsparse_add_op.cc102 const int num_dims = a_shape->dim_size(0); in Compute() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dreshape_op.cc41 const int64 num_dims = sizes_shape.num_elements(); in Compile() local
Dconv_ops.cc41 int num_dims = shape.dims(); in ExpandedFilterShapeForDepthwiseConvolution() local
192 int num_dims() const { return num_spatial_dims_ + 2; } in num_dims() function in tensorflow::__anon7321a4ba0111::ConvOp
343 int num_dims() const { return num_spatial_dims_ + 2; } in num_dims() function in tensorflow::__anon7321a4ba0111::ConvBackpropInputOp
501 int num_dims() const { return num_spatial_dims_ + 2; } in num_dims() function in tensorflow::__anon7321a4ba0111::ConvBackpropFilterOp
Dpooling_ops.cc63 int num_dims() const { return num_spatial_dims_ + 2; } in num_dims() function in tensorflow::__anonfe107e1c0111::PoolingOp
286 int num_dims() const { return num_spatial_dims_ + 2; } in num_dims() function in tensorflow::__anonfe107e1c0111::MaxPoolGradOp
406 int num_dims() const { return num_spatial_dims_ + 2; } in num_dims() function in tensorflow::__anonfe107e1c0111::AvgPoolGradOp
Dextract_image_patches_op.cc37 const int num_dims = ksizes_.size(); in Compile() local
/external/tensorflow/tensorflow/python/kernel_tests/
Dreduce_join_op_test.py33 def _input_array(num_dims): argument
47 def _joined_array(num_dims, reduce_dim): argument
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/
Dtypes.h31 inline bool NextIndex(const int num_dims, const int* dims, int* current) { in NextIndex()
56 inline size_t ReducedOutputOffset(const int num_dims, const int* dims, in ReducedOutputOffset()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dparallel_loop_emitter.cc38 const int64 num_dims = shape_.dimensions_size(); in EmitIndexAndSetExitBasicBlock() local
Dconv_canonicalization.cc45 const int64 num_dims = num_spatial_dims + 2; in Run() local
/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/
Dscatter_add_ndim_op.cc69 const int32 num_dims = in Compute() local
/external/tensorflow/tensorflow/contrib/factorization/kernels/
Dclustering_ops_test.cc50 Graph* SetUpKmeansPlusPlusInitialization(int num_dims, int num_points, in SetUpKmeansPlusPlusInitialization()
175 Graph* SetUpNearestNeighbors(int num_dims, int num_points, int num_centers, in SetUpNearestNeighbors()
/external/tensorflow/tensorflow/contrib/lite/java/src/main/native/
Dtensor_jni.cc212 int num_dims = tensor->dims->size; in Java_org_tensorflow_lite_Tensor_readMultiDimensionalArray() local
234 int num_dims = tensor->dims->size; in Java_org_tensorflow_lite_Tensor_shape() local
/external/tensorflow/tensorflow/contrib/factorization/python/ops/
Dgmm_test.py73 def make_random_centers(num_centers, num_dims): argument
/external/tensorflow/tensorflow/contrib/grid_rnn/python/ops/
Dgrid_rnn_cell.py51 num_dims=1, argument
568 def _parse_rnn_config(num_dims, ls_input_dims, ls_output_dims, ls_priority_dims, argument

123