| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_conv2d_impl.h | 56 Eigen::array<Eigen::IndexPair<int64>, 1> contract_dims; in EigenConvImpl() local
57 contract_dims[0] = Eigen::IndexPair<int64>(1, 0); in EigenConvImpl()
87 .contract(kernel.reshape(kernel_dims), contract_dims) in EigenConvImpl()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_backward_spatial_convolutions.h | 244 array<IndexPair<TensorIndex>, 1> contract_dims; variable
247 contract_dims[0] = IndexPair<TensorIndex>(0, 0);
250 contract_dims[0] = IndexPair<TensorIndex>(1, 1);
290 contract_dims)
303 contract_dims)
539 array<IndexPair<TensorIndex>, 1> contract_dims; variable
540 contract_dims[0] = IndexPair<TensorIndex>(1, 0);
596 contract_dims),
603 .contract(input_shuffled, contract_dims))
|
| D | eigen_backward_cuboid_convolutions.h | 244 array<IndexPair<TensorIndex>, 1> contract_dims; variable
247 contract_dims[0] = IndexPair<TensorIndex>(0, 0);
250 contract_dims[0] = IndexPair<TensorIndex>(1, 1);
287 contract_dims)
299 contract_dims)
536 array<IndexPair<TensorIndex>, 1> contract_dims; variable
537 contract_dims[0] = IndexPair<TensorIndex>(1, 0);
594 contract_dims),
602 .contract(the_kernel, contract_dims))
|
| D | eigen_mkldnn_contraction_kernel_test.cc | 178 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> contract_dims; in TEST() local
179 contract_dims[0].first = 1; in TEST()
180 contract_dims[0].second = 0; in TEST()
182 Tensor2dQInt32 res = lhs.contract(rhs, contract_dims); in TEST()
|
| D | conv_grad_input_ops.h | 691 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> contract_dims;
692 contract_dims[0].first = 1;
693 contract_dims[0].second = 1;
703 C.device(context->eigen_cpu_device()) = A.contract(B, contract_dims);
|
| D | conv_grad_ops_3d.cc | 488 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> contract_dims; in Compute() local
489 contract_dims[0].first = 1; in Compute()
490 contract_dims[0].second = 1; in Compute()
500 C.device(context->eigen_cpu_device()) = A.contract(B, contract_dims); in Compute()
947 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> contract_dims; in Compute() local
948 contract_dims[0].first = 0; in Compute()
949 contract_dims[0].second = 0; in Compute()
996 C.device(context->eigen_cpu_device()) += A.contract(B, contract_dims); in Compute()
|
| D | conv_grad_filter_ops.cc | 556 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> contract_dims; in Compute() local
557 contract_dims[0].first = 0; in Compute()
558 contract_dims[0].second = 0; in Compute()
593 C.device(context->eigen_cpu_device()) += A.contract(B, contract_dims); in Compute()
|
| D | eigen_spatial_convolutions-inl.h | 1641 array<IndexPair<TensorIndex>, 1> contract_dims;
1642 contract_dims[0] = IndexPair<TensorIndex>(1, 0);
1738 contract_dims, output_kernel)
1748 .contract(kernel.reshape(kernel_dims), contract_dims, output_kernel)
1759 contract_dims, output_kernel)
1766 .contract(kernel.reshape(kernel_dims), contract_dims, output_kernel)
|
| D | eigen_spatial_convolutions_test.cc | 1323 array<DimPair, 2> contract_dims; in TEST() local
1324 contract_dims[0] = DimPair(0, 0); in TEST()
1325 contract_dims[1] = DimPair(2, 1); in TEST()
1349 contract_dims) in TEST()
1360 contract_dims) in TEST()
|
| D | eigen_cuboid_convolution.h | 1944 array<IndexPair<TensorIndex>, 1> contract_dims;
1945 contract_dims[0] = IndexPair<TensorIndex>(1, 0);
1981 contract_dims)
1988 .contract(kernel.reshape(kernel_dims), contract_dims)
|
| /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
107 contract_dims.push_back(dim); in ProcessDotArg()
120 return TransposeReshape(arg, loc, outer_dims, contract_dims, shape, in ProcessDotArg()
124 return TransposeReshape(arg, loc, contract_dims, outer_dims, shape, rewriter); in ProcessDotArg()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | eigen_spatial_convolutions.h | 134 array<IndexPair<TensorIndex>, 1> contract_dims; variable
135 contract_dims[0] = IndexPair<TensorIndex>(1, 0);
218 .contract(kernel.reshape(kernel_dims), contract_dims)
|
| /external/ruy/ruy/ |
| D | test.h | 1065 Eigen::array<DimPair, 1> contract_dims(
1081 tensor_lhs.contract(tensor_rhs, contract_dims);
1084 (tensor_lhs.contract(tensor_rhs, contract_dims) +
1093 tensor_lhs.contract(tensor_rhs, contract_dims);
1095 tensor_dst.device(device) = tensor_lhs.contract(tensor_rhs, contract_dims)
|