Home
last modified time | relevance | path

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

/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorConvolutionSycl.h29 …ename Evaluator, typename CoeffReturnType, typename KernelType, typename Index, typename InputDims,
32 …ename Evaluator, typename CoeffReturnType, typename KernelType, typename Index, typename InputDims,
34 struct EigenConvolutionKernel<Evaluator, CoeffReturnType, KernelType, Index, InputDims, Kernel_acce…
42 internal::IndexMapper<Index, InputDims, 1, Evaluator::Layout> indexMapper;
47 internal::IndexMapper<Index, InputDims, 1, Evaluator::Layout> indexMapper_,
99 …ename Evaluator, typename CoeffReturnType, typename KernelType, typename Index, typename InputDims,
101 struct EigenConvolutionKernel<Evaluator, CoeffReturnType, KernelType, Index, InputDims, Kernel_acce…
109 internal::IndexMapper<Index, InputDims, 2, Evaluator::Layout> indexMapper;
114 internal::IndexMapper<Index, InputDims, 2, Evaluator::Layout> indexMapper_,
184 …ename Evaluator, typename CoeffReturnType, typename KernelType, typename Index, typename InputDims,
[all …]
DTensorConvolution.h24 template <typename Index, typename InputDims, int NumKernelDims, int Layout>
27 IndexMapper(const InputDims& input_dims, const array<Index, NumKernelDims>& kernel_dims, in IndexMapper()
209 static const int NumDims = internal::array_size<InputDims>::value;
579 template <typename InputEvaluator, typename Index, typename InputDims,
583 const internal::IndexMapper<Index, InputDims, 1, InputEvaluator::Layout>
631 template <typename InputEvaluator, typename Index, typename InputDims,
635 const internal::IndexMapper<Index, InputDims, 2, InputEvaluator::Layout>
703 template <typename InputEvaluator, typename Index, typename InputDims>
706 const internal::IndexMapper<Index, InputDims, 3, InputEvaluator::Layout>
880 typedef typename TensorEvaluator<InputArgType, GpuDevice>::Dimensions InputDims;
[all …]
DTensorReduction.h70 template <typename InputDims, typename ReducedDims> EIGEN_DEVICE_FUNC
71 static void run(const InputDims& input_dims,
72 const array<bool, internal::array_size<InputDims>::value>& reduced,
74 const int NumInputDims = internal::array_size<InputDims>::value;
90 template <typename InputDims, typename Index, size_t Rank> EIGEN_DEVICE_FUNC
91 static void run(const InputDims& input_dims, const array<bool, Rank>&,
93 const int NumInputDims = internal::array_size<InputDims>::value;
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dtranspose.h118 absl::Span<int64_t const> InputDims() const { return original_a_dims_; } in InputDims() function