Home
last modified time | relevance | path

Searched refs:IndexPair (Results 1 – 25 of 38) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/
Deigen_backward_spatial_convolutions.h65 IndexPair<typename internal::traits<OutputBackward>::Index>, 1>,
84 IndexPair<typename internal::traits<OutputBackward>::Index>, 1>,
243 array<IndexPair<TensorIndex>, 1> contract_dims;
246 contract_dims[0] = IndexPair<TensorIndex>(0, 0);
249 contract_dims[0] = IndexPair<TensorIndex>(1, 1);
331 const array<IndexPair<typename internal::traits<Input>::Index>, 1>,
341 const array<IndexPair<typename internal::traits<Input>::Index>, 1>,
454 array<IndexPair<TensorIndex>, 1> contract_dims;
457 contract_dims[0] = IndexPair<TensorIndex>(1, 1);
460 contract_dims[0] = IndexPair<TensorIndex>(0, 0);
Deigen_backward_cuboid_convolutions.h54 IndexPair<typename internal::traits<OutputBackward>::Index>, 1>,
73 IndexPair<typename internal::traits<OutputBackward>::Index>, 1>,
244 array<IndexPair<TensorIndex>, 1> contract_dims;
247 contract_dims[0] = IndexPair<TensorIndex>(0, 0);
250 contract_dims[0] = IndexPair<TensorIndex>(1, 1);
338 IndexPair<typename internal::traits<Input>::Index>, 1>,
369 IndexPair<typename internal::traits<Input>::Index>, 1>,
536 array<IndexPair<TensorIndex>, 1> contract_dims;
537 contract_dims[0] = IndexPair<TensorIndex>(1, 0);
Dpad_op.h34 Eigen::array<Eigen::IndexPair<Tpadding>, Dims> paddings, in operator()
50 Eigen::array<Eigen::IndexPair<Tpadding>, 0>, T) {
Dmatmul_op.cc77 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair, in ExplicitVectorMatrixOptimization()
107 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair, in ExplicitVectorMatrixOptimization()
122 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair, in launch()
255 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair, in launch()
464 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in Compute()
539 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair) { in operator ()()
552 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair) { in operator ()()
Deigen_attention_test.cc38 std::vector<IndexPair<float>> offsets; in TEST()
74 std::vector<IndexPair<float>> offsets; in TEST()
Dattention_ops.cc111 std::vector<Eigen::IndexPair<float> > offset_vec; in Compute()
118 offset_vec.push_back(Eigen::IndexPair<float>(offset_x, offset_y)); in Compute()
Deigen_attention.h57 const std::vector<IndexPair<float> >& offsets, in GlimpseExtractionOp()
242 const std::vector<IndexPair<float> > offsets_;
256 const std::vector<IndexPair<float> >& offsets, const bool normalized = true,
Dsdca_internal.cc53 const Eigen::array<Eigen::IndexPair<int>, 1> product_dims = { in UpdateDenseDeltaWeights()
54 Eigen::IndexPair<int>(1, 0)}; in UpdateDenseDeltaWeights()
214 const Eigen::array<Eigen::IndexPair<int>, 1> product_dims = { in ComputeWxAndWeightedExampleNorm()
215 Eigen::IndexPair<int>(1, 1)}; in ComputeWxAndWeightedExampleNorm()
Dbatch_matmul_op_impl.h62 Eigen::IndexPair<Eigen::DenseIndex> ContractionDims(bool adj_x, bool adj_y) { in ContractionDims()
63 return Eigen::IndexPair<Eigen::DenseIndex>(adj_x ? 0 : 1, adj_y ? 1 : 0); in ContractionDims()
88 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> contract_pairs; in Run()
118 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> contract_pairs;
508 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> contract_pairs;
Dconv_2d.h47 const Eigen::array<Eigen::IndexPair<IndexType>, Dims>& pad_dims, in operator()
141 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair,
254 Eigen::array<Eigen::IndexPair<IndexType>, NDIMS> padding;
Dpad_op.cc255 Eigen::array<Eigen::IndexPair<Tpadding>, Dims> paddings_array; in Operate()
305 Eigen::array<Eigen::IndexPair<int32>, Dims> paddings, T pad_value); \
311 Eigen::array<Eigen::IndexPair<int64>, Dims> paddings, T pad_value); \
Dconv_ops.cc89 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in operator ()()
90 dim_pair[0] = Eigen::IndexPair<Eigen::DenseIndex>(1, 0); in operator ()()
105 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in operator ()()
106 dim_pair[0] = Eigen::IndexPair<Eigen::DenseIndex>(1, 0); in operator ()()
935 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair, \
Dmatmul_op.h56 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair);
Dgemm_functors.h105 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in operator()
Dconv_ops_fused_impl.h267 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in operator()
268 dim_pair[0] = Eigen::IndexPair<Eigen::DenseIndex>(1, 0); in operator()
284 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in operator()
285 dim_pair[0] = Eigen::IndexPair<Eigen::DenseIndex>(1, 0); in operator()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dmultithreaded_conv.h64 const Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1>& dim_pair) { in operator()
100 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in operator()
101 dim_pair[0] = Eigen::IndexPair<Eigen::DenseIndex>(1, 0); in operator()
114 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> dim_pair; in operator()
115 dim_pair[0] = Eigen::IndexPair<Eigen::DenseIndex>(1, 0); in operator()
Deigen_spatial_convolutions.h73 const array<IndexPair<typename internal::traits<Input>::Index>,
86 const array<IndexPair<typename internal::traits<Input>::Index>,
134 array<IndexPair<TensorIndex>, 1> contract_dims;
135 contract_dims[0] = IndexPair<TensorIndex>(1, 0);
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Deigen_spatial_convolutions.h76 const array<IndexPair<typename internal::traits<Input>::Index>,
89 const array<IndexPair<typename internal::traits<Input>::Index>,
137 array<IndexPair<TensorIndex>, 1> contract_dims;
138 contract_dims[0] = IndexPair<TensorIndex>(1, 0);
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorMeta.h154 template <typename Idx> struct IndexPair {
155 EIGEN_CONSTEXPR EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE IndexPair() : first(0), second(0) {}
156 …EIGEN_CONSTEXPR EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE IndexPair(Idx f, Idx s) : first(f), second(s…
158 EIGEN_DEVICE_FUNC void set(IndexPair<Idx> val) {
DTensorIndexList.h56 constexpr EIGEN_DEVICE_FUNC operator IndexPair<DenseIndex>() const {
57 return IndexPair<DenseIndex>(f, s);
60 EIGEN_DEVICE_FUNC void set(const IndexPair<DenseIndex>& val) { in set()
95 EIGEN_DEVICE_FUNC void update_value(T& val, IndexPair<DenseIndex> new_val) {
99 EIGEN_DEVICE_FUNC void update_value(type2indexpair<f, s>& val, IndexPair<DenseIndex> new_val) {
336 …EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC constexpr IndexPair<DenseIndex> operator[] (const DenseIndex…
337 …::array_size<internal::IndexTuple<FirstType, OtherTypes...> >::value-1, IndexPair<DenseIndex>>::ge…
339 …EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC void set(const DenseIndex i, const IndexPair<DenseIndex> val…
340 …l::array_size<internal::IndexTuple<FirstType, OtherTypes...>>::value-1, IndexPair<DenseIndex> >::s…
DTensorContractionMapper.h152 …EIGEN_STRONG_INLINE IndexPair<Index> computeIndexPair(Index row, Index col, const Index distance) …
195 return IndexPair<Index>(linidx[0], linidx[1]);
253 const IndexPair<Index> indexPair = this->computeIndexPair(i, j, packet_size - 1);
272 const IndexPair<Index> internal_pair = this->computeIndexPair(i + k, j, 1);
/external/eigen/unsupported/test/
Dcxx11_tensor_index_list.cpp170 …typedef Eigen::IndexPairList<Eigen::type2indexpair<0,10>, Eigen::IndexPair<DenseIndex>, Eigen::typ… in test_type2indexpair_list()
171 …typedef Eigen::IndexPairList<Eigen::IndexPair<DenseIndex>, Eigen::type2indexpair<1,11>, Eigen::Ind… in test_type2indexpair_list()
177 d2_b.set(1, Eigen::IndexPair<DenseIndex>(1,11)); in test_type2indexpair_list()
180 d2_c.set(0, Eigen::IndexPair<DenseIndex>(Eigen::IndexPair<DenseIndex>(0,10))); in test_type2indexpair_list()
181 d2_c.set(1, Eigen::IndexPair<DenseIndex>(1,11)); // setting type2indexpair to correct value. in test_type2indexpair_list()
182 d2_c.set(2, Eigen::IndexPair<DenseIndex>(2,12)); in test_type2indexpair_list()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_conv2d_impl.h50 Eigen::array<Eigen::IndexPair<int64>, 1> contract_dims; in EigenConvImpl()
51 contract_dims[0] = Eigen::IndexPair<int64>(1, 0); in EigenConvImpl()
/external/tensorflow/tensorflow/contrib/rnn/kernels/
Dblas_gemm.h78 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> contract_pairs;
80 Eigen::IndexPair<Eigen::DenseIndex>(transa == false, transb == true);
/external/eigen/bench/tensors/
Dtensor_benchmarks.h185 Eigen::array<Eigen::IndexPair<TensorIndex>, 2> paddings; in padding()
186 paddings[0] = Eigen::IndexPair<TensorIndex>(0, 0); in padding()
187 paddings[1] = Eigen::IndexPair<TensorIndex>(2, 1); in padding()

12