Searched refs:dim_count (Results 1 – 8 of 8) sorted by relevance
93 int dim_count = 0; in BitcastOp_Compute() local99 dim_count = TF_NumDims(tensor); in BitcastOp_Compute()101 (dim_count > 0 && in BitcastOp_Compute()102 TF_Dim(tensor, dim_count - 1) == k->out_size / k->in_size))) { in BitcastOp_Compute()111 auto* dims = new int64_t[dim_count + 1]; in BitcastOp_Compute()112 int new_dim_count = dim_count; in BitcastOp_Compute()113 for (int dim = 0; dim < dim_count; ++dim) { in BitcastOp_Compute()
37 int dim_count) { in StridedSlicePadIndices() argument39 TFLITE_CHECK_LE(dim_count, 5); in StridedSlicePadIndices()40 TFLITE_CHECK_GE(dim_count, p->start_indices_count); in StridedSlicePadIndices()44 const int pad_count = dim_count - p->start_indices_count; in StridedSlicePadIndices()67 p->start_indices_count = dim_count; in StridedSlicePadIndices()68 p->stop_indices_count = dim_count; in StridedSlicePadIndices()69 p->strides_count = dim_count; in StridedSlicePadIndices()
25 int dim_count = pad_values.size(); in PadAttributeArray() local26 if (attribute_dim_count < dim_count) { in PadAttributeArray()27 Shape strided_slice_shape = Shape({dim_count}); in PadAttributeArray()32 for (int i = attribute_dim_count; i < dim_count; i++) { in PadAttributeArray()
141 int dim_count = 2; in test_rw_image_access_qualifier() local155 err |= clEnqueueNDRangeKernel(commands, kernel, dim_count, in test_rw_image_access_qualifier()
47 int dim_count = ElementCount(*output_dims[i]); in TestSplitVFloat() local48 for (int j = 0; j < dim_count; j++) { in TestSplitVFloat()93 int dim_count = ElementCount(*output_dims[i]); in TestSplitVFloat() local94 for (int j = 0; j < dim_count; j++) { in TestSplitVFloat()
1985 int dim_count = 0; in PropagateOnConv() local1991 dim_map[permuted_conv_dims_numbers.output_batch_dimension()] = dim_count; in PropagateOnConv()1992 new_dim_numbers.set_output_batch_dimension(dim_count++); in PropagateOnConv()1995 dim_count; in PropagateOnConv()1996 new_dim_numbers.set_output_spatial_dimensions(j, dim_count); in PropagateOnConv()1997 dim_count++; in PropagateOnConv()2000 dim_map[permuted_conv_dims_numbers.output_feature_dimension()] = dim_count; in PropagateOnConv()2001 new_dim_numbers.set_output_feature_dimension(dim_count); in PropagateOnConv()2749 int dim_count = 0; in PerformSpaceToBatchOnConvolution() local2754 dim_map[dim_numbers.output_batch_dimension()] = dim_count; in PerformSpaceToBatchOnConvolution()[all …]
111 const int dim_count = transfer_dim(xfer); in transfer_overlap() local116 for (int dim = 0; dim < dim_count; dim++) { in transfer_overlap()
1134 void PadIndices(int dim_count) {1136 CHECK_GE(dim_count, start_indices.size());1140 for (int i = start_indices.size(); i < dim_count; i++) {