/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
runtime_conv2d_impl.h | 50 Eigen::array<Eigen::IndexPair<int64>, 1> contract_dims; local 51 contract_dims[0] = Eigen::IndexPair<int64>(1, 0); 81 .contract(kernel.reshape(kernel_dims), contract_dims)
|
/external/tensorflow/tensorflow/core/kernels/ |
eigen_backward_spatial_convolutions.h | 243 array<IndexPair<TensorIndex>, 1> contract_dims; local 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; local 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) [all...] |
eigen_backward_cuboid_convolutions.h | 244 array<IndexPair<TensorIndex>, 1> contract_dims; local 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; local 537 contract_dims[0] = IndexPair<TensorIndex>(1, 0); 594 contract_dims), 602 .contract(the_kernel, contract_dims))
|
conv_grad_ops_3d.cc | 477 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> contract_dims; variable 478 contract_dims[0].first = 1; 479 contract_dims[0].second = 1; 489 C.device(context->eigen_cpu_device()) = A.contract(B, contract_dims); 936 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> contract_dims; variable [all...] |
conv_grad_filter_ops.cc | 357 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> contract_dims; variable 358 contract_dims[0].first = 0; 359 contract_dims[0].second = 0; 394 C.device(context->eigen_cpu_device()) += A.contract(B, contract_dims); [all...] |
conv_grad_input_ops.cc | 473 Eigen::array<Eigen::IndexPair<Eigen::DenseIndex>, 1> contract_dims; variable 474 contract_dims[0].first = 1; 475 contract_dims[0].second = 1; 485 C.device(context->eigen_cpu_device()) = A.contract(B, contract_dims); [all...] |
eigen_spatial_convolutions-inl.h | [all...] |
eigen_spatial_convolutions_test.cc | 1323 array<DimPair, 2> contract_dims; local [all...] |
eigen_cuboid_convolution.h | [all...] |
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
eigen_spatial_convolutions.h | 137 array<IndexPair<TensorIndex>, 1> contract_dims; local 138 contract_dims[0] = IndexPair<TensorIndex>(1, 0); 221 .contract(kernel.reshape(kernel_dims), contract_dims)
|
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
eigen_spatial_convolutions.h | 134 array<IndexPair<TensorIndex>, 1> contract_dims; local 135 contract_dims[0] = IndexPair<TensorIndex>(1, 0); 218 .contract(kernel.reshape(kernel_dims), contract_dims)
|