| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_backward_spatial_convolutions.h | 243 array<IndexPair<TensorIndex>, 1> contract_dims; variable
246 contract_dims[0] = IndexPair<TensorIndex>(0, 0);
249 contract_dims[0] = IndexPair<TensorIndex>(1, 1);
284 contract_dims)
296 contract_dims)
454 array<IndexPair<TensorIndex>, 1> contract_dims; variable
457 contract_dims[0] = IndexPair<TensorIndex>(1, 1);
460 contract_dims[0] = IndexPair<TensorIndex>(0, 0);
488 contract_dims)
496 .contract(output_backward.reshape(output_dims), 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 | conv_grad_ops_3d.cc | 477 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> contract_dims; in Compute() local
478 contract_dims[0].first = 1; in Compute()
479 contract_dims[0].second = 1; in Compute()
489 C.device(context->eigen_cpu_device()) = A.contract(B, contract_dims); in Compute()
936 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> contract_dims; in Compute() local
937 contract_dims[0].first = 0; in Compute()
938 contract_dims[0].second = 0; in Compute()
985 C.device(context->eigen_cpu_device()) += A.contract(B, contract_dims); in Compute()
|
| D | conv_grad_filter_ops.cc | 357 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> contract_dims; in Compute() local
358 contract_dims[0].first = 0; in Compute()
359 contract_dims[0].second = 0; in Compute()
394 C.device(context->eigen_cpu_device()) += A.contract(B, contract_dims); in Compute()
|
| D | conv_grad_input_ops.cc | 473 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> contract_dims; in Compute() local
474 contract_dims[0].first = 1; in Compute()
475 contract_dims[0].second = 1; in Compute()
485 C.device(context->eigen_cpu_device()) = A.contract(B, contract_dims); in Compute()
|
| 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_spatial_convolutions-inl.h | 1398 array<IndexPair<TensorIndex>, 1> contract_dims;
1399 contract_dims[0] = IndexPair<TensorIndex>(1, 0);
1484 contract_dims, output_kernel)
1490 .contract(kernel.reshape(kernel_dims), contract_dims, output_kernel)
|
| D | eigen_cuboid_convolution.h | 1750 array<IndexPair<TensorIndex>, 1> contract_dims;
1751 contract_dims[0] = IndexPair<TensorIndex>(1, 0);
1787 contract_dims)
1794 .contract(kernel.reshape(kernel_dims), contract_dims)
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_conv2d_impl.h | 50 Eigen::array<Eigen::IndexPair<int64>, 1> contract_dims; in EigenConvImpl() local
51 contract_dims[0] = Eigen::IndexPair<int64>(1, 0); in EigenConvImpl()
81 .contract(kernel.reshape(kernel_dims), contract_dims) in EigenConvImpl()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | eigen_spatial_convolutions.h | 137 array<IndexPair<TensorIndex>, 1> contract_dims; variable
138 contract_dims[0] = IndexPair<TensorIndex>(1, 0);
221 .contract(kernel.reshape(kernel_dims), contract_dims)
|
| /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)
|