Searched refs:kNumDims (Results 1 – 3 of 3) sorted by relevance
240 const int kNumDims = 3; in MakeSparseTensor() local242 Tensor indices(DT_INT64, TensorShape({total_nnz, kNumDims})); in MakeSparseTensor()244 Tensor shape(DT_INT64, TensorShape({kNumDims})); in MakeSparseTensor()
1205 const int kNumDims = dim.size(); in CreateBlockedMemDescHelper() local1206 mkldnn_dim_t* input_dims = new mkldnn_dim_t[kNumDims]; in CreateBlockedMemDescHelper()1207 mkldnn_dim_t* input_strides = new mkldnn_dim_t[kNumDims]; in CreateBlockedMemDescHelper()1208 for (int i = 0; i < kNumDims; ++i) { in CreateBlockedMemDescHelper()1213 mkldnn_memory_desc_init_by_strides(blocked_md, kNumDims, input_dims, in CreateBlockedMemDescHelper()
3227 const int kNumDims = 5; in ConvertConv3DHelper() local3248 if (weights_drsck.shape_.nbDims != kNumDims) { in ConvertConv3DHelper()3260 if (tf_dilations.size() != kNumDims) { in ConvertConv3DHelper()3283 if (tf_stride.size() != kNumDims) { in ConvertConv3DHelper()3417 const int kNumDims = 5; in ConvertPool3D() local3445 if (tf_stride.size() != kNumDims) { in ConvertPool3D()3456 if (tf_kernel.size() != kNumDims) { in ConvertPool3D()