Home
last modified time | relevance | path

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

/aosp_15_r20/external/mesa3d/src/intel/vulkan/grl/gpu/
H A Dqbvh6.h160 const int4 exp_i = (int4)(qnode->exp[0], qnode->exp[1], qnode->exp[2], 0.0f); in extractAABB_QBVHNodeN() local
211 const int4 exp_i = (int4)(qnode->exp[0], qnode->exp[1], qnode->exp[2], 0.0f); in getAABB_QBVHNodeN() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/sparse/
H A Dkernels_gpu.cu.cc360 const T exp_i = Eigen::numext::exp(ldg(logits + r_i) - row_max); in CalculateRowSoftmax() local
/aosp_15_r20/external/mesa3d/src/intel/vulkan/grl/include/
H A DGRLGen12.h582 const int4 exp_i = { node->exp_x, node->exp_y, node->exp_z, 0 }; in InternalNode_GetChildAABB() local
/aosp_15_r20/external/pytorch/test/
H A Dtest_sparse.py642 def test_tensor(x, exp_i, exp_v): argument
723 def test_tensor(x, exp_i, exp_v): argument