Home
last modified time | relevance | path

Searched refs:NewDimension (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Deigen_cuboid_convolution.h58 template <typename NewDimension, Index Planes, Index Rows, Index Cols,
64 TensorEvaluator<const TensorReshapingOp<NewDimension,
75 NewDimension, const TensorVolumePatchOp<
84 NewDimension, const TensorVolumePatchOp<
98 NewDimension, in TensorContractionInputMapper()
342 NewDimension, const TensorVolumePatchOp<
680 template <typename NewDimension, Index Planes, Index Rows, Index Cols,
686 TensorEvaluator<const TensorReshapingOp<NewDimension,
699 NewDimension, const TensorVolumePatchOp<
708 NewDimension, const TensorVolumePatchOp<
[all …]
Deigen_spatial_convolutions-inl.h55 template <typename NewDimension, Index Rows, Index Cols, typename ArgType,
62 const TensorReshapingOp<NewDimension,
74 NewDimension, const TensorImagePatchOp<Rows, Cols, ArgType> >,
84 NewDimension, const TensorImagePatchOp<Rows, Cols, ArgType> >,
98 NewDimension, const TensorImagePatchOp<Rows, Cols, ArgType> >, in TensorContractionInputMapper()
275 NewDimension, const TensorImagePatchOp<Rows, Cols, ArgType> >,
510 template <typename NewDimension, Index Rows, Index Cols, typename ArgType,
517 const TensorReshapingOp<NewDimension,
530 NewDimension, const TensorImagePatchOp<Rows, Cols, ArgType> >,
540 NewDimension, const TensorImagePatchOp<Rows, Cols, ArgType> >,
[all …]
Deigen_spatial_convolutions.h31 template <typename NewDimension, Index Rows, Index Cols, typename ArgType,
41 NewDimension, const TensorImagePatchOp<Rows, Cols, ArgType> >,
50 NewDimension, const TensorImagePatchOp<Rows, Cols, ArgType> >,
Deigen_spatial_convolutions_test.cc1404 using NewDimension = Eigen::DSizes<Index, 2>; in PackRhsHelper() typedef
1418 NewDimension, const TensorImagePatchOp<Dynamic, Dynamic, ArgType>>, in PackRhsHelper()
1490 NewDimension reshape_dims; in PackRhsHelper()
1495 TensorReshapingOp<NewDimension, decltype(image_patch_op)>( in PackRhsHelper()
1577 using NewDimension = Eigen::DSizes<Index, 2>; in PackLhsHelper() typedef
1590 TensorEvaluator<const TensorReshapingOp<NewDimension, ArgType>, in PackLhsHelper()
1626 NewDimension reshape_dims; in PackLhsHelper()
1662 TensorReshapingOp<NewDimension, ArgType>(tensor_map, reshape_dims); in PackLhsHelper()