Home
last modified time | relevance | path

Searched defs:result_dims (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/eigen/unsupported/test/
H A Dcxx11_tensor_convolution_sycl.cpp46 Eigen::array<IndexType, 3> result_dims = {{outdim0, outdim1, outdim2}}; in test_larg_expr1D() local
107 Eigen::array<IndexType, 3> result_dims = {{outdim0, outdim1, outdim2}}; in test_larg_expr2D() local
168 Eigen::array<IndexType, 3> result_dims = {{outdim0, outdim1, outdim2}}; in test_larg_expr3D() local
223 Eigen::array<IndexType, 2> result_dims = {{2, 3}}; in test_evals() local
269 Eigen::array<IndexType, 2> result_dims = {{2, 2}}; in test_expr() local
H A Dcxx11_tensor_contract_sycl.cpp51 Eigen::array<IndexType, 2> result_dims = {{m_size, n_size}}; in test_sycl_contraction() local
166 Eigen::array<IndexType, 2> result_dims = {{m_size, n_size}}; in test_no_out_of_bounds() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dresampler_ops.cc200 std::vector<int64_t> result_dims, in BoundSamples()
581 std::vector<int64_t> result_dims(warp_dims.begin(), warp_dims.end() - 1); in Compile() local
664 std::vector<int64_t> result_dims(warp_dims.begin(), warp_dims.end() - 1); in Compile() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Ddot_op_emitter.cc1267 DimensionVector result_dims; in CollapseFirstNDims() local