Home
last modified time | relevance | path

Searched refs:num_dims (Results 1 – 25 of 60) sorted by relevance

123

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dconv_canonicalization.cc45 const int64 num_dims = num_spatial_dims + 2; in Run() local
58 std::vector<int64> new_input_dim_order(num_dims); in Run()
59 std::vector<int64> new_input_dims(num_dims); in Run()
67 new_input_dim_order[num_dims - 1] = input_feature_dim; in Run()
68 new_input_dims[num_dims - 1] = in Run()
79 std::vector<int64> new_kernel_dim_order(num_dims); in Run()
80 std::vector<int64> new_kernel_dims(num_dims); in Run()
86 new_kernel_dim_order[num_dims - 2] = kernel_input_feature_dim; in Run()
87 new_kernel_dims[num_dims - 2] = in Run()
89 new_kernel_dim_order[num_dims - 1] = kernel_output_feature_dim; in Run()
[all …]
Dparallel_loop_emitter.cc38 const int64 num_dims = shape_.dimensions_size(); in EmitIndexAndSetExitBasicBlock() local
39 llvm_ir::IrArray::Index array_index(num_dims); in EmitIndexAndSetExitBasicBlock()
44 const int bounds_index = num_dims - 1 - i; in EmitIndexAndSetExitBasicBlock()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dpooling_ops.cc44 OP_REQUIRES(ctx, ksize_int.size() == num_dims(), in PoolingOp()
47 num_dims(), " dimensions")); in PoolingOp()
49 OP_REQUIRES(ctx, stride_int.size() == num_dims(), in PoolingOp()
52 num_dims(), " dimensions")); in PoolingOp()
53 for (int i = 0; i < num_dims(); ++i) { in PoolingOp()
63 int num_dims() const { return num_spatial_dims_ + 2; } in num_dims() function in tensorflow::__anonfe107e1c0111::PoolingOp
90 OP_REQUIRES(ctx, ksize_shape.num_elements() == num_dims(), in Compile()
93 num_dims(), " dimensions")); in Compile()
102 OP_REQUIRES(ctx, stride_shape.num_elements() == num_dims(), in Compile()
105 num_dims(), " dimensions")); in Compile()
[all …]
Dextract_image_patches_op.cc37 const int num_dims = ksizes_.size(); in Compile() local
40 ctx, num_dims >= 3, in Compile()
42 const int num_spatial_dims = num_dims - 2; in Compile()
44 OP_REQUIRES(ctx, strides_.size() == num_dims, in Compile()
47 num_dims, " dimensions")); in Compile()
48 OP_REQUIRES(ctx, dilations_.size() == num_dims, in Compile()
51 num_dims, " dimensions")); in Compile()
53 int batch_dim = GetTensorBatchDimIndex(num_dims, data_format); in Compile()
54 int feature_dim = GetTensorFeatureDimIndex(num_dims, data_format); in Compile()
70 int input_dim = GetTensorSpatialDimIndex(num_dims, data_format, i); in Compile()
[all …]
Dconv_ops.cc41 int num_dims = shape.dims(); in ExpandedFilterShapeForDepthwiseConvolution() local
42 CHECK_GE(num_dims, 2); in ExpandedFilterShapeForDepthwiseConvolution()
44 expanded_shape.set_dim(num_dims - 1, shape.dim_size(num_dims - 2) * in ExpandedFilterShapeForDepthwiseConvolution()
45 shape.dim_size(num_dims - 1)); in ExpandedFilterShapeForDepthwiseConvolution()
192 int num_dims() const { return num_spatial_dims_ + 2; } in num_dims() function in tensorflow::__anon7321a4ba0111::ConvOp
195 OP_REQUIRES(ctx, strides_.size() == num_dims(), in Compile()
198 num_dims(), " dimensions")); in Compile()
199 int batch_dim = GetTensorBatchDimIndex(num_dims(), data_format_); in Compile()
200 int feature_dim = GetTensorFeatureDimIndex(num_dims(), data_format_); in Compile()
206 OP_REQUIRES(ctx, dilations_.size() == num_dims(), in Compile()
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/
Dreduce_join_op_test.py33 def _input_array(num_dims): argument
42 formatter = "{:0%db}" % num_dims
43 strings = [formatter.format(i) for i in xrange(2**num_dims)]
44 return np.array(strings, dtype="S%d" % num_dims).reshape([2] * num_dims)
47 def _joined_array(num_dims, reduce_dim): argument
57 formatter = "{:0%db}" % (num_dims - 1)
58 result = np.zeros(shape=[2] * (num_dims - 1), dtype="S%d" % (2 * num_dims))
60 for i in xrange(2**(num_dims - 1)):
79 num_dims = 3
80 truth = ["{:03b}".format(i) for i in xrange(2**num_dims)]
[all …]
/external/tensorflow/tensorflow/core/util/
Dtensor_format.h91 inline int GetTensorSpatialDims(int num_dims, TensorFormat format) { in GetTensorSpatialDims() argument
93 return num_dims - 3; // Exclude N,C,InnerC. in GetTensorSpatialDims()
95 return num_dims - 2; // Exclude N,C. in GetTensorSpatialDims()
99 inline int GetFilterTensorSpatialDims(int num_dims, FilterTensorFormat format) { in GetFilterTensorSpatialDims() argument
101 return num_dims - 3; // Exclude O,I,InnerI. in GetFilterTensorSpatialDims()
103 return num_dims - 2; // Exclude O,I. in GetFilterTensorSpatialDims()
130 inline int GetTensorBatchDimIndex(int num_dims, TensorFormat format) { in GetTensorBatchDimIndex() argument
145 inline int GetTensorFeatureDimIndex(int num_dims, TensorFormat format) { in GetTensorFeatureDimIndex() argument
148 return num_dims - 1; in GetTensorFeatureDimIndex()
159 inline int GetTensorInnerFeatureDimIndex(int num_dims, TensorFormat format) { in GetTensorInnerFeatureDimIndex() argument
[all …]
Dtensor_format_test.cc160 int num_dims = GetTensorDimsFromSpatialDims(num_spatial_dims, format); in RunDimensionIndexesTest() local
162 << ", num_dims=" << num_dims; in RunDimensionIndexesTest()
163 EXPECT_EQ(GetTensorBatchDimIndex(num_dims, format), tdm.n()); in RunDimensionIndexesTest()
165 EXPECT_EQ(GetTensorFeatureDimIndex(num_dims, format), tdm.c()); in RunDimensionIndexesTest()
168 EXPECT_EQ(GetTensorSpatialDimIndex(num_dims, format, i), tdm.spatial(i)); in RunDimensionIndexesTest()
176 int num_dims = GetFilterTensorDimsFromSpatialDims(num_spatial_dims, format); in RunDimensionIndexesTest() local
178 << ", num_dims=" << num_dims; in RunDimensionIndexesTest()
179 EXPECT_EQ(GetFilterTensorOutputChannelsDimIndex(num_dims, format), fdm.o()); in RunDimensionIndexesTest()
181 EXPECT_EQ(GetFilterTensorInputChannelsDimIndex(num_dims, format), fdm.i()); in RunDimensionIndexesTest()
184 EXPECT_EQ(GetFilterTensorSpatialDimIndex(num_dims, format, i), in RunDimensionIndexesTest()
/external/tensorflow/tensorflow/contrib/grid_rnn/python/ops/
Dgrid_rnn_cell.py51 num_dims=1, argument
116 if num_dims < 1:
117 raise ValueError('dims must be >= 1: {}'.format(num_dims))
119 self._config = _parse_rnn_config(num_dims, input_dims, output_dims,
133 self._cells = [my_cell_fn()] * num_dims
135 self._cells = [my_cell_fn() for _ in range(num_dims)]
182 new_output = [None] * conf.num_dims
183 new_state = [None] * conf.num_dims
237 c_prev = [None] * conf.num_dims
238 m_prev = [None] * conf.num_dims
[all …]
/external/tensorflow/tensorflow/core/kernels/
Droll_op.cc39 const int num_dims, const gtl::ArraySlice<int>& dim_size, in DoRoll() argument
42 auto work = [input, output, num_dims, &dim_size, &threshold, &dim_range]( in DoRoll()
45 gtl::InlinedVector<int, 4> indices(num_dims); in DoRoll()
48 for (int i = 0; i < num_dims; i++) { in DoRoll()
65 for (int j = num_dims - 1; j >= 0; j--) { in DoRoll()
102 const int num_dims, const gtl::ArraySlice<int>& dim_size, in DoRollWithMemcpy() argument
107 auto work = [input, output, num_dims, &dim_size, &threshold, &dim_range, isd]( in DoRollWithMemcpy()
131 gtl::InlinedVector<int, 4> indicies(num_dims); in DoRollWithMemcpy()
135 for (int i = 0; i < num_dims; i++) { in DoRollWithMemcpy()
153 for (int i = num_dims - 1; i > isd; i--) indicies[i] = 0; in DoRollWithMemcpy()
[all …]
Dconv_grad_ops.cc103 const int num_dims = num_spatial_dims + 2; in ConvBackpropComputeDimensionsV2() local
104 if (input_shape.dims() != num_dims) { in ConvBackpropComputeDimensionsV2()
105 return errors::InvalidArgument(label, ": input must be ", num_dims, in ConvBackpropComputeDimensionsV2()
108 if (filter_shape.dims() != num_dims) { in ConvBackpropComputeDimensionsV2()
109 return errors::InvalidArgument(label, ": filter must be ", num_dims, in ConvBackpropComputeDimensionsV2()
112 if (out_backprop_shape.dims() != num_dims) { in ConvBackpropComputeDimensionsV2()
113 return errors::InvalidArgument(label, ": out_backprop must be ", num_dims, in ConvBackpropComputeDimensionsV2()
116 int batch_dim = GetTensorBatchDimIndex(num_dims, data_format); in ConvBackpropComputeDimensionsV2()
126 int feature_dim = GetTensorFeatureDimIndex(num_dims, data_format); in ConvBackpropComputeDimensionsV2()
130 if (dims->in_depth != filter_shape.dim_size(num_dims - 2)) { in ConvBackpropComputeDimensionsV2()
[all …]
Dnth_element_op.cc52 const int num_dims = input_in.dims(); in Compute() local
53 OP_REQUIRES(context, num_dims >= 1, in Compute()
58 context, input_in.dim_size(num_dims - 1) > n, in Compute()
63 n = input_in.dim_size(num_dims - 1) - n - 1; in Compute()
68 for (int i = 0; i < num_dims - 1; ++i) { in Compute()
Ddiag_op.cc49 const int num_dims = diagonal.dims(); in Compute() local
51 context, 0 != num_dims, in Compute()
54 for (int i = 0; i < num_dims; ++i) { in Compute()
57 for (int i = 0; i < num_dims; ++i) { in Compute()
79 const int num_dims = tensor.dims(); in Compute() local
80 const int out_dims = num_dims / 2; in Compute()
81 OP_REQUIRES(context, 0 == num_dims % 2, in Compute()
Dsparse_to_dense_op.cc60 const int64 num_dims = indices.dims() > 1 ? indices.dim_size(1) : 1; in Compute() local
68 OP_REQUIRES(c, output_shape.NumElements() == num_dims, in Compute()
71 output_shape.NumElements(), " should be: ", num_dims)); in Compute()
96 TensorShape ix_shape({num_elems, num_dims}); in Compute()
Dsparse_sparse_binary_op_shared.cc66 typename TTypes<T>::ConstFlat b_values, int64 b_nnz, int num_dims, in UnionSparseIndicesAndValues() argument
77 num_dims)) { in UnionSparseIndicesAndValues()
183 const int num_dims = a_indices_t->dim_size(1); in Compute() local
189 b_values, b_nnz, num_dims, &a_augmented_values, in Compute()
196 ctx->allocate_output(0, TensorShape({sum_nnz, num_dims}), in Compute()
/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
333 if (num_dims > 0) { in Java_org_tensorflow_Tensor_allocate()
347 int64_t* dims_copy = new int64_t[num_dims]; in Java_org_tensorflow_Tensor_allocate()
348 for (int i = 0; i < num_dims; ++i) { in Java_org_tensorflow_Tensor_allocate()
352 num_dims, static_cast<size_t>(sizeInBytes)); in Java_org_tensorflow_Tensor_allocate()
391 size_t nonScalarTF_STRINGTensorSize(JNIEnv* env, jarray value, int num_dims) { in nonScalarTF_STRINGTensorSize() argument
392 if (num_dims == 0) { in nonScalarTF_STRINGTensorSize()
403 ret += nonScalarTF_STRINGTensorSize(env, elem, num_dims - 1); in nonScalarTF_STRINGTensorSize()
408 void fillNonScalarTF_STRINGTensorData(JNIEnv* env, jarray value, int num_dims, in fillNonScalarTF_STRINGTensorData() argument
411 if (num_dims == 0) { in fillNonScalarTF_STRINGTensorData()
[all …]
Doperation_jni.cc106 jsize num_dims = TF_GraphGetTensorNumDims(graph, output, status); in Java_org_tensorflow_Operation_shape() local
111 if (num_dims < 0) return nullptr; in Java_org_tensorflow_Operation_shape()
120 std::unique_ptr<int64_t[]> cdims(new int64_t[num_dims]); in Java_org_tensorflow_Operation_shape()
121 TF_GraphGetTensorShape(graph, output, cdims.get(), static_cast<int>(num_dims), in Java_org_tensorflow_Operation_shape()
129 jlongArray ret = env->NewLongArray(num_dims); in Java_org_tensorflow_Operation_shape()
131 for (int i = 0; i < num_dims; ++i) { in Java_org_tensorflow_Operation_shape()
Doperation_builder_jni.cc247 jint num_dims) { in Java_org_tensorflow_OperationBuilder_setAttrShape() argument
253 if (num_dims > 0) { in Java_org_tensorflow_OperationBuilder_setAttrShape()
254 cvalue.reset(new int64_t[num_dims]); in Java_org_tensorflow_OperationBuilder_setAttrShape()
256 for (int i = 0; i < num_dims; ++i) { in Java_org_tensorflow_OperationBuilder_setAttrShape()
262 TF_SetAttrShape(d, cname, cvalue.get(), static_cast<int>(num_dims)); in Java_org_tensorflow_OperationBuilder_setAttrShape()
268 jintArray num_dims) { in Java_org_tensorflow_OperationBuilder_setAttrShapeList() argument
274 const int num_dims_length = env->GetArrayLength(num_dims); in Java_org_tensorflow_OperationBuilder_setAttrShapeList()
286 static_cast<jint*>(env->GetPrimitiveArrayCritical(num_dims, nullptr)); in Java_org_tensorflow_OperationBuilder_setAttrShapeList()
294 env->ReleasePrimitiveArrayCritical(num_dims, num_dims_elems, JNI_ABORT); in Java_org_tensorflow_OperationBuilder_setAttrShapeList()
/external/tensorflow/tensorflow/contrib/factorization/kernels/
Dclustering_ops_test.cc50 Graph* SetUpKmeansPlusPlusInitialization(int num_dims, int num_points, in SetUpKmeansPlusPlusInitialization() argument
54 Tensor points(DT_FLOAT, TensorShape({num_points, num_dims})); in SetUpKmeansPlusPlusInitialization()
73 template <int num_points, int num_to_sample, int num_dims,
77 testing::ItemsProcessed(static_cast<int64>(iters) * num_points * num_dims * in BM_KmeansPlusPlusInitialization()
81 num_dims, num_points, num_to_sample, retries_per_sample); in BM_KmeansPlusPlusInitialization()
134 template <int num_points, int num_to_sample, int num_dims>
137 testing::ItemsProcessed(static_cast<int64>(iters) * num_points * num_dims * in BM_KMC2Initialization()
175 Graph* SetUpNearestNeighbors(int num_dims, int num_points, int num_centers, in SetUpNearestNeighbors() argument
178 Tensor points(DT_FLOAT, TensorShape({num_points, num_dims})); in SetUpNearestNeighbors()
179 Tensor centers(DT_FLOAT, TensorShape({num_centers, num_dims})); in SetUpNearestNeighbors()
[all …]
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/
Dtypes.h31 inline bool NextIndex(const int num_dims, const int* dims, int* current) { in NextIndex() argument
32 TFLITE_DCHECK_GT(num_dims, 0); in NextIndex()
36 for (int idx = num_dims - 1; idx >= 0; --idx) { in NextIndex()
56 inline size_t ReducedOutputOffset(const int num_dims, const int* dims, in ReducedOutputOffset() argument
59 TFLITE_DCHECK_GT(num_dims, 0); in ReducedOutputOffset()
63 for (int idx = 0; idx < num_dims; ++idx) { in ReducedOutputOffset()
/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/
Dscatter_add_ndim_op.cc69 const int32 num_dims = in Compute() local
75 for (int32 i = 0; i < input_tensor.shape().dims() - num_dims; ++i) { in Compute()
76 num_data_per_index *= input_tensor.shape().dim_size(num_dims + i); in Compute()
86 for (int32 j = 0; j < num_dims; j++) { in Compute()
95 for (int32 j = 0; j < num_dims; j++) { in Compute()
/external/tensorflow/tensorflow/contrib/factorization/python/ops/
Dgmm_test.py58 self.num_dims = 2
62 self.num_dims)
73 def make_random_centers(num_centers, num_dims): argument
75 np.random.rand(num_centers, num_dims).astype(np.float32) * 500)
79 num_centers, num_dims = centers.shape
82 np.random.randn(num_points, num_dims).astype(np.float32) * 20)
104 self.assertAllEqual(list(clusters.shape), [self.num_centers, self.num_dims])
/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
213 if (num_dims == 0) { in Java_org_tensorflow_lite_Tensor_readMultiDimensionalArray()
219 num_dims, static_cast<jarray>(value)); in Java_org_tensorflow_lite_Tensor_readMultiDimensionalArray()
234 int num_dims = tensor->dims->size; in Java_org_tensorflow_lite_Tensor_shape() local
235 jintArray result = env->NewIntArray(num_dims); in Java_org_tensorflow_lite_Tensor_shape()
237 for (int i = 0; i < num_dims; ++i) { in Java_org_tensorflow_lite_Tensor_shape()
/external/tensorflow/tensorflow/contrib/lite/g3doc/
Dcustom_operators.md45 int num_dims = NumDimensions(input);
47 TfLiteIntArray* output_size = TfLiteIntArrayCreate(num_dims);
48 for (int i=0; i<num_dims; ++i) {
64 int num_dims = NumDimensions(input);
65 for (int i = 0; i < num_dims; ++i) {
/external/tensorflow/tensorflow/python/client/
Dtf_session_helper.cc335 int num_dims = TF_GraphGetTensorNumDims(graph, output, out_status); in TF_GraphGetTensorShapeHelper() local
340 if (num_dims == -1) { in TF_GraphGetTensorShapeHelper()
346 if (num_dims == 0) { in TF_GraphGetTensorShapeHelper()
350 result.resize(num_dims); in TF_GraphGetTensorShapeHelper()
351 TF_GraphGetTensorShape(graph, output, result.data(), num_dims, out_status); in TF_GraphGetTensorShapeHelper()
475 int num_dims, in TF_GraphGetTensorShape_wrapper() argument
477 std::vector<int64_t> dims(num_dims); in TF_GraphGetTensorShape_wrapper()
478 TF_GraphGetTensorShape(graph, output, dims.data(), num_dims, status); in TF_GraphGetTensorShape_wrapper()

123