Searched refs:NestedTensor_elementwise_Tensor (Results 1 – 2 of 2) sorted by relevance
75 Tensor NestedTensor_elementwise_Tensor( in NestedTensor_elementwise_Tensor() function186 return NestedTensor_elementwise_Tensor( in NestedTensor_add_Tensor()196 return NestedTensor_elementwise_Tensor( in NestedTensor_sub_Tensor()203 return NestedTensor_elementwise_Tensor( in NestedTensor_mul_Tensor()215 return NestedTensor_elementwise_Tensor( in NestedTensor_div_Tensor()229 return NestedTensor_elementwise_Tensor( in NestedTensor_masked_fill()299 return NestedTensor_elementwise_Tensor( in ge_scalar_nested()307 return NestedTensor_elementwise_Tensor( in gt_scalar_nested()315 return NestedTensor_elementwise_Tensor( in eq_scalar_nested()324 return NestedTensor_elementwise_Tensor( in eq_tensor_nested()
24 …edTensors (functions that can be efficiently implemented via NestedTensor_elementwise_Tensor which…