Searched defs:exp_i (Results 1 – 4 of 4) sorted by relevance
/aosp_15_r20/external/mesa3d/src/intel/vulkan/grl/gpu/ |
H A D | qbvh6.h | 160 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 D | kernels_gpu.cu.cc | 360 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 D | GRLGen12.h | 582 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 D | test_sparse.py | 642 def test_tensor(x, exp_i, exp_v): argument 723 def test_tensor(x, exp_i, exp_v): argument
|