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.h60 template <typename NewDimension, Index Planes, Index Rows, Index Cols,
66 TensorEvaluator<const TensorReshapingOp<NewDimension,
77 NewDimension, const TensorVolumePatchOp<
86 NewDimension, const TensorVolumePatchOp<
100 NewDimension, in TensorContractionInputMapper()
344 NewDimension, const TensorVolumePatchOp<
873 template <typename NewDimension, Index Planes, Index Rows, Index Cols,
879 TensorEvaluator<const TensorReshapingOp<NewDimension,
892 NewDimension, const TensorVolumePatchOp<
901 NewDimension, const TensorVolumePatchOp<
[all …]
Deigen_spatial_convolutions-inl.h57 template <typename NewDimension, Index Rows, Index Cols, typename ArgType,
64 const TensorReshapingOp<NewDimension,
76 NewDimension, const TensorImagePatchOp<Rows, Cols, ArgType> >,
86 NewDimension, const TensorImagePatchOp<Rows, Cols, ArgType> >,
102 NewDimension, const TensorImagePatchOp<Rows, Cols, ArgType> >, in TensorContractionInputMapper()
281 NewDimension, const TensorImagePatchOp<Rows, Cols, ArgType> >,
675 template <typename NewDimension, Index Rows, Index Cols, typename ArgType,
682 const TensorReshapingOp<NewDimension,
695 NewDimension, const TensorImagePatchOp<Rows, Cols, ArgType> >,
705 NewDimension, const TensorImagePatchOp<Rows, Cols, ArgType> >,
[all …]
Deigen_spatial_convolutions.h77 template <typename NewDimension, Index Rows, Index Cols, typename ArgType,
87 NewDimension, const TensorImagePatchOp<Rows, Cols, ArgType> >,
96 NewDimension, const TensorImagePatchOp<Rows, Cols, ArgType> >,
Deigen_spatial_convolutions_test.cc1405 using NewDimension = Eigen::DSizes<Index, 2>; in PackRhsHelper() typedef
1419 NewDimension, const TensorImagePatchOp<Dynamic, Dynamic, ArgType>>, in PackRhsHelper()
1511 NewDimension reshape_dims; in PackRhsHelper()
1516 TensorReshapingOp<NewDimension, decltype(image_patch_op)>( in PackRhsHelper()
1600 using NewDimension = Eigen::DSizes<Index, 2>; in PackLhsHelper() typedef
1613 TensorEvaluator<const TensorReshapingOp<NewDimension, ArgType>, in PackLhsHelper()
1649 NewDimension reshape_dims; in PackLhsHelper()
1689 TensorReshapingOp<NewDimension, ArgType>(tensor_map, reshape_dims); in PackLhsHelper()