Searched refs:first_tensor (Results 1 – 7 of 7) sorted by relevance
71 const Tensor& first_tensor = tensors[0]; in can_use_native_serial_stack_impl() local75 if (dim >= first_tensor.dim()) return false; in can_use_native_serial_stack_impl()77 if (first_tensor.numel() == 0 && first_tensor.dim() == 1) return false; in can_use_native_serial_stack_impl()79 if (result.dtype() != first_tensor.dtype()) return false; in can_use_native_serial_stack_impl()81 auto first_tensor_mem_format = first_tensor.suggest_memory_format(); in can_use_native_serial_stack_impl()82 ScalarType dtype = first_tensor.scalar_type(); in can_use_native_serial_stack_impl()95 auto const &first_tensor_shape = first_tensor.sizes(); in can_use_native_serial_stack_impl()99 TORCH_CHECK(tensors[i].sizes() == first_tensor.sizes(), in can_use_native_serial_stack_impl()107 tensor.strides() != first_tensor.strides() || in can_use_native_serial_stack_impl()116 int64_t numel_in_stack = first_tensor.numel() * tensors.size(); in can_use_native_serial_stack_impl()
181 first_tensor = next((item for item in tensor if item is not None), None)182 if first_tensor is None:185 ir.TensorType(_torch_dtype_to_onnx_compatible_dtype(first_tensor.dtype))279 first_tensor = _get_first_tensor_in_node_list(arg)280 assert first_tensor is not None282 … arg = ir.SequenceType(_get_type_from_tensor(first_tensor)) # type: ignore[assignment]
141 first_tensor = True173 if first_tensor:216 first_tensor = False
212 int first_tensor = -1, last_tensor = -1, first_tensor_list = -1, in verify() local218 if (first_tensor == -1) { in verify()219 first_tensor = arg.index(); in verify()
1123 TfLiteTensor* first_tensor = context->tensors; in TEST() local1128 EXPECT_EQ(first_tensor, context->tensors); in TEST()1143 TfLiteTensor* first_tensor = context->tensors; in TEST() local1151 EXPECT_NE(first_tensor, context->tensors); in TEST()
1435 first_tensor = g.op(1440 first_tensor_shape = g.op("Shape", first_tensor)
849 const auto& first_tensor = inputs[0]; in inputsAreScalars() local850 return first_tensor.sizes()[0] == 1 && first_tensor.dim() == 1; in inputsAreScalars()