Searched defs:contract_dims (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_backward_spatial_convolutions.h | 225 array<IndexPair<TensorIndex>, 1> contract_dims; variable
520 array<IndexPair<TensorIndex>, 1> contract_dims; variable
|
| D | eigen_backward_cuboid_convolutions.h | 244 array<IndexPair<TensorIndex>, 1> contract_dims; variable
536 array<IndexPair<TensorIndex>, 1> contract_dims; variable
|
| D | eigen_mkldnn_contraction_kernel_test.cc | 178 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> contract_dims; in TEST() local
|
| D | conv_grad_ops_3d.cc | 541 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> contract_dims; in Compute() local
1057 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> contract_dims; in Compute() local
|
| D | conv_grad_filter_ops.cc | 573 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> contract_dims; in Compute() local
|
| D | eigen_spatial_convolutions_test.cc | 1323 array<DimPair, 2> contract_dims; in TEST() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_conv_impl.h | 53 Eigen::array<Eigen::IndexPair<Eigen::Index>, 1> contract_dims; in EigenConv2DImpl() local
151 Eigen::array<Eigen::IndexPair<Eigen::Index>, 1> contract_dims; in EigenConv3DImpl() local
|