Home
last modified time | relevance | path

Searched refs:dim_count (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/c/kernels/
Dbitcast_op.cc93 int dim_count = 0; in BitcastOp_Compute() local
99 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()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dstrided_slice_logic.h37 int dim_count) { in StridedSlicePadIndices() argument
39 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()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_strided_slice_attributes.cc25 int dim_count = pad_values.size(); in PadAttributeArray() local
26 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()
/external/OpenCL-CTS/test_conformance/basic/
Dtest_rw_image_access_qualifier.cpp141 int dim_count = 2; in test_rw_image_access_qualifier() local
155 err |= clEnqueueNDRangeKernel(commands, kernel, dim_count, in test_rw_image_access_qualifier()
/external/tensorflow/tensorflow/lite/micro/kernels/
Dsplit_v_test.cc47 int dim_count = ElementCount(*output_dims[i]); in TestSplitVFloat() local
48 for (int j = 0; j < dim_count; j++) { in TestSplitVFloat()
93 int dim_count = ElementCount(*output_dims[i]); in TestSplitVFloat() local
94 for (int j = 0; j < dim_count; j++) { in TestSplitVFloat()
/external/tensorflow/tensorflow/compiler/xla/service/
Dspace_to_batch_converter.cc1985 int dim_count = 0; in PropagateOnConv() local
1991 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() local
2754 dim_map[dim_numbers.output_batch_dimension()] = dim_count; in PerformSpaceToBatchOnConvolution()
[all …]
/external/mesa3d/src/gallium/drivers/virgl/
Dvirgl_transfer_queue.c111 const int dim_count = transfer_dim(xfer); in transfer_overlap() local
116 for (int dim = 0; dim < dim_count; dim++) { in transfer_overlap()
/external/tensorflow/tensorflow/lite/toco/
Dmodel.h1134 void PadIndices(int dim_count) {
1136 CHECK_GE(dim_count, start_indices.size());
1140 for (int i = start_indices.size(); i < dim_count; i++) {