Searched defs:contract_dims (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_backward_spatial_convolutions.h | 244 array<IndexPair<TensorIndex>, 1> contract_dims; variable
539 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 | 488 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> contract_dims; in Compute() local
947 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> contract_dims; in Compute() local
|
| D | conv_grad_filter_ops.cc | 556 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/lite/kernels/internal/optimized/ |
| D | eigen_spatial_convolutions.h | 134 array<IndexPair<TensorIndex>, 1> contract_dims; variable
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_conv2d_impl.h | 56 Eigen::array<Eigen::IndexPair<int64>, 1> contract_dims; in EigenConvImpl() local
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | lower_general_dot.cc | 105 llvm::SmallVector<int64_t, 5> contract_dims; in ProcessDotArg() local
|