/external/tensorflow/tensorflow/core/util/ |
D | tensor_format.h | 91 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()
|
D | tensor_format_test.cc | 160 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/ |
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() 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/ |
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 | 39 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
|
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 | attention_ops.cc | 45 const int32 num_dims = input_shape.dims(); 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
|
D | sparse_add_op.cc | 102 const int num_dims = a_shape->dim_size(0); in Compute() local
|
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
D | reshape_op.cc | 41 const int64 num_dims = sizes_shape.num_elements(); in Compile() local
|
D | conv_ops.cc | 41 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
|
D | pooling_ops.cc | 63 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
|
D | extract_image_patches_op.cc | 37 const int num_dims = ksizes_.size(); in Compile() local
|
/external/tensorflow/tensorflow/python/kernel_tests/ |
D | reduce_join_op_test.py | 33 def _input_array(num_dims): argument 47 def _joined_array(num_dims, reduce_dim): argument
|
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/ |
D | types.h | 31 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/ |
D | parallel_loop_emitter.cc | 38 const int64 num_dims = shape_.dimensions_size(); in EmitIndexAndSetExitBasicBlock() local
|
D | conv_canonicalization.cc | 45 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
|
/external/tensorflow/tensorflow/contrib/factorization/kernels/ |
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()
|
/external/tensorflow/tensorflow/contrib/lite/java/src/main/native/ |
D | tensor_jni.cc | 212 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/ |
D | gmm_test.py | 73 def make_random_centers(num_centers, num_dims): argument
|
/external/tensorflow/tensorflow/contrib/grid_rnn/python/ops/ |
D | grid_rnn_cell.py | 51 num_dims=1, argument 568 def _parse_rnn_config(num_dims, ls_input_dims, ls_output_dims, ls_priority_dims, argument
|